Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Bo Jin 0003

dblp:74/3468-3 · DBLP profile ↗
← Back
44ranked-venue papers
0as first author
32since 2021 · last 2026
0000-0003-1028-5989ORCID · conflict

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 35 · 26 since 2021Applied, interdisciplinary, general and emerging computing · 11 · 9 since 2021Databases, data management, data science and information retrieval · 9 · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 9 · 4 since 2021Systems, architecture and hardware · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
16 papers
Reinforcement learning · 57% Multi-agent systems · 16% Planning, search and constraint satisfaction · 11%
Computer architecture, parallel and distributed computing, and storage systems
3 papers
Cloud and datacenter computing · 94% Distributed systems · 6%
Interdisciplinary, comprehensive, and emerging computing
4 papers
Medical and health informatics · 74% Computational social science and digital humanities · 20% Computational finance and economics · 6%

Topics — the 30 heaviest of 45, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Reinforcement learning
multi-agent reinforcement learning
3.962026
Learning Roles With Emergent Social Value Orientations · IEEE Trans. Pattern Anal. Mach. Intell. 2026
F2A2: Flexible Fully-decentralized Approximate Actor-critic for Cooperative Multi-agent Reinforcement Learning · J. Mach. Learn. Res. 2023
Learning Cooperative Oversubscription for Cloud by Chance-Constrained Multi-Agent Reinforcement Learning · WWW 2023
Knowledge, reasoning and agents › Planning, search and constraint satisfaction
multi-agent path finding
1.422025
SkyRover: A Modular Simulator for Cross-Domain Pathfinding · IJCAI 2025
Multi-Agent Path Finding with Prioritized Communication Learning · ICRA 2022
Cloud and datacenter computing › resource management
cloud resource management
1.222023
Learning Cooperative Oversubscription for Cloud by Chance-Constrained Multi-Agent Reinforcement Learning · WWW 2023
VMAgent: A Practical Virtual Machine Scheduling Platform · IJCAI 2022
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › planning › agent planning
embodied planning
1.012026
On-policy Reinforcement Fine-tuning with Offline reward for Multi-step Embodied Planning · ACL (1) 2026
Knowledge, reasoning and agents › Multi-agent systems › game theory
social dilemmas
1.012026
Learning Roles With Emergent Social Value Orientations · IEEE Trans. Pattern Anal. Mach. Intell. 2026
Machine learning › Reinforcement learning
deep reinforcement learning
0.912025
Reward Translation via Reward Machine in Semi-Alignable MDPs · ICML 2025
Robotics › Robot manipulation
embodied foundation models
0.912025
Generative Multi-Agent Collaboration in Embodied AI: A Systematic Review · IJCAI 2025
Knowledge, reasoning and agents › Multi-agent systems
embodied multi-agent system
0.912025
Generative Multi-Agent Collaboration in Embodied AI: A Systematic Review · IJCAI 2025
Machine learning › Reinforcement learning
reward design
0.912025
Reward Translation via Reward Machine in Semi-Alignable MDPs · ICML 2025
Machine learning › Reinforcement learning › multi-agent reinforcement learning
cooperative multi-agent reinforcement learning
0.712023
F2A2: Flexible Fully-decentralized Approximate Actor-critic for Cooperative Multi-agent Reinforcement Learning · J. Mach. Learn. Res. 2023
Machine learning › Reinforcement learning › actor-critic methods
decentralized actor-critic
0.712023
F2A2: Flexible Fully-decentralized Approximate Actor-critic for Cooperative Multi-agent Reinforcement Learning · J. Mach. Learn. Res. 2023
Machine learning › Generative modeling
diffusion model
0.712023
Hierarchical Diffusion for Offline Decision Making · ICML 2023
Machine learning › Reinforcement learning › hierarchical reinforcement learning
hierarchical offline reinforcement learning
0.712023
Hierarchical Diffusion for Offline Decision Making · ICML 2023
Machine learning › Reinforcement learning
offline reinforcement learning
0.712023
Hierarchical Diffusion for Offline Decision Making · ICML 2023
Cloud and datacenter computing › resource management
resource oversubscription
0.712023
Learning Cooperative Oversubscription for Cloud by Chance-Constrained Multi-Agent Reinforcement Learning · WWW 2023
Machine learning › Reinforcement learning › multi-agent reinforcement learning
cooperative reinforcement learning
0.612022
Structured Cooperative Reinforcement Learning With Time-Varying Composite Action Space · IEEE Trans. Pattern Anal. Mach. Intell. 2022
Machine learning › Reinforcement learning › markov decision process
factored action spaces
0.612022
Structured Cooperative Reinforcement Learning With Time-Varying Composite Action Space · IEEE Trans. Pattern Anal. Mach. Intell. 2022
Machine learning › Reinforcement learning › multi-agent reinforcement learning
non-stationarity
0.612022
Dealing with Non-Stationarity in MARL via Trust-Region Decomposition · ICLR 2022
Cloud and datacenter computing
cluster resource management and scheduling
0.612022
VMAgent: A Practical Virtual Machine Scheduling Platform · IJCAI 2022
Cloud and datacenter computing › virtualization › virtual machine management
virtual machine scheduling
0.612022
VMAgent: A Practical Virtual Machine Scheduling Platform · IJCAI 2022
Machine learning › Reinforcement learning
meta-reinforcement learning
0.512021
HMRL: Hyper-Meta Learning for Sparse Reward Reinforcement Learning Problem · KDD 2021
Machine learning › Reinforcement learning › reward design
reward shaping
0.512021
HMRL: Hyper-Meta Learning for Sparse Reward Reinforcement Learning Problem · KDD 2021
Machine learning › Reinforcement learning
sparse reward reinforcement learning
0.512021
HMRL: Hyper-Meta Learning for Sparse Reward Reinforcement Learning Problem · KDD 2021
Medical and health informatics
clinical decision support
0.312018
Learning to Detect Pathogenic Microorganism of Community-acquired Pneumonia · SIGIR 2018
Medical and health informatics
electronic health records
0.312018
Learning to Detect Pathogenic Microorganism of Community-acquired Pneumonia · SIGIR 2018
Computer vision › Vision and language
vision-language model
0.312026
On-policy Reinforcement Fine-tuning with Offline reward for Multi-step Embodied Planning · ACL (1) 2026
Machine learning › Reinforcement learning
markov decision process
0.312025
Reward Translation via Reward Machine in Semi-Alignable MDPs · ICML 2025
Robotics › Motion planning and robot control
path planning
0.312025
SkyRover: A Modular Simulator for Cross-Domain Pathfinding · IJCAI 2025
Robotics › Legged, aerial and field robots › aerial robots
unmanned aerial vehicle
0.312025
SkyRover: A Modular Simulator for Cross-Domain Pathfinding · IJCAI 2025
Machine learning › Probabilistic and Bayesian machine learning › stochastic processes
point process
0.212016
Modeling Contagious Merger and Acquisition via Point Processes with a Profile Regression Prior · IJCAI 2016

Methods — techniques the papers use, named apart from their topics

multi-agent reinforcement learning · 3.4chance-constrained optimization · 1.3reinforcement fine-tuning · 1.0rank regularizer · 1.0offline reward · 1.0mutual information maximization · 1.0simulation · 0.9reward machines · 0.9large language model · 0.9graph matching · 0.9foundation model · 0.9point process · 0.8theory-of-mind · 0.7primal-dual hybrid gradient descent · 0.7parameter sharing · 0.7actor-critic · 0.7reinforcement learning · 0.6operations research · 0.6
YearPublicationVenuePosition
2026 On-policy Reinforcement Fine-tuning with Offline reward for Multi-step Embodied Planning
abstract
Embodied planning requires agents to make coherent multi-step decisions based on dynamic visual observations and verbal goals.While recent vision-language models (VLMs) excel at static perception tasks, they struggle in interactive environments.Reinforcement learning (RL) offers a natural way to address this limitation, yet online RL approaches suffer from costly interaction and sparse rewards in embodied settings.This paper introduces ORBIT, an On-policy Reinforcement finetuning (RFT) framework with offline rewards for EmBodIed Task Planning, that preserves the generalization benefits of RFT while addressing the challenges of costly interaction and sparse rewards, supported by solid theoretical guarantees.Our approach is evaluated on EmbodiedBench, a recent benchmark for interactive embodied tasks, covering both in-domain and out-of-domain scenarios.Experimental results show that ORBIT achieves SOTA performance on EB-ALFRED, outperforming all closed-source and online-RL-based methods, while being substantially more efficient in training speed and computational cost, remaining robust to sub-optimal expert trajectories, and exhibiting strong generalization to unseen environments.We released all code and data at https://github.com/mail-taii/Reinforced- Reasoning-for
Chloe Gu, Guanbo Wang, Wenhao Li 0001, Bo Jin 0003
ACL (1)7
2026 Learning Roles With Emergent Social Value Orientations
abstract
Social dilemmas can be considered situations where individual rationality leads to collective irrationality. The multi-agent reinforcement learning community has leveraged ideas from social science, such as social value orientations (SVO), to solve social dilemmas in complex cooperative tasks. In this paper, we first introduce the typical "division of labor or roles" mechanism in human society, and provide a promising solution for intertemporal social dilemmas (ISD) with SVOs. A novel learning framework, called Learning Roles with Emergent SVOs (RESVO), is proposed to transform the learning of roles into the social value orientation emergence, which is symmetrically solved by endowing agents with altruism to share rewards with other agents. An SVO-based role embedding space is then constructed by individual conditioning policies on roles with a novel rank regularizer and mutual information maximizer. Experiments show that RESVO achieves a stable division of labor and cooperation in ISDs with different complexity.
Wenhao Li 0001, Xiangfeng Wang 0001, Bo Jin 0003, Hongyuan Zha
IEEE Trans. Pattern Anal. Mach. Intell.3
2025 Multi-Agent Credit Assignment with Pretrained Language Models
abstract
The difficulty of appropriately assigning credit is particularly heightened in cooperative MARL with sparse reward, due to the concurrent time and structural scales involved. Automatic subgoal generation (ASG) has recently emerged as a viable MARL approach inspired by utilizing subgoals in intrinsically motivated reinforcement learning. However, end-to-end learning of complex task planning from sparse rewards without prior knowledge, undoubtedly requires massive training samples. Moreover, the diversity-promoting nature of existing ASG methods can lead to the "over-representation" of subgoals, generating numerous spurious subgoals of limited relevance to the actual task reward and thus decreasing the sample efficiency of the algorithm. To address this problem and inspired by the disentangled representation learning, we propose a novel "disentangled" decision-making method, Semantically Aligned task decomposition in MARL (SAMA), that prompts pretrained language models with chain-of-thought that can suggest potential goals, provide suitable goal decomposition and subgoal allocation as well as self-reflection-based replanning. Additionally, SAMA incorporates language-grounded MARL to train each agent’s subgoal-conditioned policy. SAMA demonstrates considerable advantages in sample efficiency compared to state-of-the-art ASG methods, as evidenced by its performance on two challenging sparse-reward tasks, Overcooked and MiniRTS. The code is available at \url{https://anonymous.4open.science/r/SAMA/.}
Wenhao Li 0001, Baoxiang Wang 0001, Xiangfeng Wang 0001, Hao Shen 0002, Bo Jin 0003, Hongyuan Zha
AISTATS7
2025 Reward Translation via Reward Machine in Semi-Alignable MDPs
abstract
Addressing reward design complexities in deep reinforcement learning is facilitated by knowledge transfer across different domains. To this end, we define reward translation to describe the cross-domain reward transfer problem. However, current methods struggle with non-pairable and non-time-alignable incompatible MDPs. This paper presents an adaptable reward translation framework neural reward translation featuring semi-alignable MDPs, which allows efficient reward translation under relaxed constraints while handling the intricacies of incompatible MDPs. Given the inherent difficulty of directly mapping semi-alignable MDPs and transferring rewards, we introduce an indirect mapping method through reward machines, created using limited human input or LLM-based automated learning. Graph-matching techniques establish links between reward machines from distinct environments, thus enabling cross-domain reward translation within semi-alignable MDP settings. This broadens the applicability of DRL across multiple domains. Experiments substantiate our approach’s effectiveness in tasks under environments with semi-alignable MDPs.
Yun Hua, Wenhao Li 0001, Bo Jin 0003, Baoxiang Wang 0001, Hongyuan Zha, Xiangfeng Wang 0002
ICML4
2025 Dynamic Conservative Degree Allocation for Offline Multi-Agent Reinforcement Learning
Yun Hua, Junjie Sheng, Wenhao Li 0001, Bo Jin 0003, Xiangfeng Wang 0001
AAMAS5
2025 Negotiated Reasoning: On Provably Addressing Relative Over-Generalization
Junjie Sheng, Wenhao Li 0001, Bo Jin 0003, Hongyuan Zha, Jun Wang 0006, Xiangfeng Wang 0001
AAMAS3
2025 SkyRover: A Modular Simulator for Cross-Domain Pathfinding
abstract
Unmanned Aerial Vehicles (UAVs) and Automated Guided Vehicles (AGVs) increasingly collaborate in logistics, surveillance, inspection tasks and etc. However, existing simulators often focus on a single domain, limiting cross-domain study. This paper presents the SkyRover, a modular simulator for UAV-AGV multi-agent pathfinding (MAPF). SkyRover supports realistic agent dynamics, configurable 3D environments, and convenient APIs for external solvers and learning methods. By unifying ground and aerial operations, it facilitates cross-domain algorithm design, testing, and benchmarking. Experiments highlight SkyRover’s capacity for efficient pathfinding and high-fidelity simulations in UAV-AGV coordination. We believe the SkyRover fills a key gap in MAPF research. Project is available at https://sites.google.com/view/mapf3d/home.
Wenhui Ma, Wenhao Li 0001, Bo Jin 0003, Changhong Lu, Xiangfeng Wang 0001
IJCAI3
2025 Generative Multi-Agent Collaboration in Embodied AI: A Systematic Review
abstract
Embodied multi-agent systems (EMAS) have attracted growing attention for their potential to address complex, real-world challenges in areas such as logistics and robotics. Recent advances in foundation models pave the way for generative agents capable of richer communication and adaptive problem-solving. This survey provides a systematic examination of how EMAS can benefit from these generative capabilities. We propose a taxonomy that categorizes EMAS by system architectures and embodiment modalities, emphasizing how collaboration spans both physical and virtual contexts. Central building blocks, perception, planning, communication, and feedback, are then analyzed to illustrate how generative techniques bolster system robustness and flexibility. Through concrete examples, we demonstrate the transformative effects of integrating foundation models into embodied, multi-agent frameworks. Finally, we discuss challenges and future directions, underlining the significant promise of EMAS to reshape the landscape of AI-driven collaboration.
Xian Wei, Guang Chen 0001, Hao Shen 0002, Bo Jin 0003
IJCAI5
2025 LOPT: Learning Optimal Pigovian Tax in Sequential Social Dilemmas
abstract
Multi-agent reinforcement learning (MARL) has emerged as a powerful framework for modeling autonomous agents that independently optimize their individual objectives. However, in mixed-motive MARL environments, rational self-interested behaviors often lead to collectively suboptimal outcomes situations commonly referred to as social dilemmas. A key challenge in addressing social dilemmas lies in accurately quantifying and representing them in a numerical form that captures how self-interested agent behaviors impact social welfare. To address this challenge, \textit{externalities} in the economic concept is adopted and extended to denote the unaccounted-for impact of one agent's actions on others, as a means to rigorously quantify social dilemmas. Based on this measurement, a novel method, \textbf{L}earning \textbf{O}ptimal \textbf{P}igovian \textbf{T}ax (\textbf{LOPT}) is proposed. Inspired by Pigovian taxes, which are designed to internalize externalities by imposing cost on negative societal impacts, LOPT employs an auxiliary tax agent that learns an optimal Pigovian tax policy to reshape individual rewards aligned with social welfare, thereby promoting agent coordination and mitigating social dilemmas. We support LOPT with theoretical analysis and validate it on standard MARL benchmarks, including Escape Room and Cleanup. Results show that by effectively internalizing externalities that quantify social dilemmas, LOPT aligns individual objectives with collective goals, significantly improving social welfare over state-of-the-art baselines.
Yun Hua, Wenhao Li 0001, Bo Jin 0003, Xiangfeng Wang 0001, Hongyuan Zha
NeurIPS5
2025 Interpretable Hybrid-Rule Temporal Point Processes
Yunyang Cao, Juekai Lin, Hongye Wang, Wenhao Li 0001, Bo Jin 0003
ECML/PKDD (3)5
2024 Semi-supervised Learning to Defer Algorithm for Lung Disease Diagnosis
abstract
Recent research highlights the advantages of leveraging complementary strengths of both human expert and model in decision-making processes. Learning to Defer(L2D) is proposed to build a system consisting of both and improve the performance of expert or model alone. For each instance, L2D algorithms search for the optimal decision-maker between model and human expert to improve the human-ai system accuracy. However, most previous work is based on the assumption that human predictions are available for every instance, which is unrealistic in practical scenarios due to the high expense of manual annotation. To address this, we consider L2D problem where human predictions are available only for a subset of image. We propose a multistep framework for this scenario and integrate a consistency regularization loss to learn expert capability from limited human predictions. The consistency regularization loss is designed to encourage the expert to learn intrinsic information and make similar predictions for similar instances. Empirical validation on real-world data from airspace opacity diagnosis shows that the proposed framework not only outperforms several competitive baselines but also enhances the performance of various L2D algorithms under constraints of minimal human predictions.
Haoqing Chen, Bo Jin 0003, Xiangfeng Wang 0001
IEEE Big Data2
2024 FPPL: An Efficient and Non-IID Robust Federated Continual Learning Framework
abstract
Federated continual learning (FCL) aims to learn from sequential data stream in the decentralized federated learning setting, while simultaneously mitigating the catastrophic forgetting issue in classical continual learning. Existing FCL methods usually employ typical rehearsal mechanisms, which could result in privacy violations or additional onerous storage and computational burdens. In this work, an efficient and non-IID robust federated continual learning framework, called Federated Prototype-Augmented Prompt Learning (FPPL), is proposed. The FPPL can collaboratively learn lightweight prompts augmented by prototypes without rehearsal. On the client side, a fusion function is employed to fully leverage the knowledge contained in task-specific prompts for alleviating catastrophic forgetting. Additionally, global prototypes aggregated from the server are used to obtain unified representation through contrastive learning, mitigating the impact of non-IID-derived data heterogeneity. On the server side, locally uploaded prototypes are utilized to perform debiasing on the classifier, further alleviating the performance degradation caused by both non-IID and catastrophic forgetting. Empirical evaluations demonstrate the effectiveness of FPPL, achieving notable performance with an efficient design while remaining robust to diverse non-IID degrees. Code is available at: https://github.com/ycheoo/FPPL.
Yuchen He 0007, Chuyun Shen, Xiangfeng Wang 0001, Bo Jin 0003
IEEE Big Data4
2024 Optimizing Efficiency and Effectiveness in Sequential Prompt Strategy for SAM Using Reinforcement Learning
Chuyun Shen, Wenhao Li 0001, Xiangfeng Wang 0001, Bo Jin 0003, Haibin Cai
MICCAI (8)5
2024 Complementary information mutual learning for multimodality medical image segmentation
Chuyun Shen, Wenhao Li 0001, Haoqing Chen, Xiaoling Wang 0004, Fengping Zhu, Xiangfeng Wang 0001, Bo Jin 0003
Neural Networks8
2023 Hierarchical Diffusion for Offline Decision Making
abstract
Offline reinforcement learning typically introduces a hierarchical structure to solve the long-horizon problem so as to address its thorny issue of variance accumulation. Problems of deadly triad, limited data and reward sparsity, however, still remain, rendering the design of effective, hierarchical offline RL algorithms for general-purpose policy learning a formidable challenge. In this paper, we first formulate the problem of offline long-horizon decision-$\mathbf{M}$ak$\mathbf{I}$ng from the perspective of conditional generative modeling by incorporating goals into the control-as-inference graphic models. A $\mathbf{H}$ierarchical trajectory-level $\mathbf{D}$iffusion probabilistic model is then proposed with classifier-free guidance. HDMI employs a cascade framework that utilizes the reward-conditional goal diffuser for the subgoal discovery and the goal-conditional trajectory diffuser for generating the corresponding action sequence of subgoals. Planning-based subgoal extraction and transformer-based diffusion are employed to deal with the sub-optimal data pollution and long-range subgoal dependencies in the goal diffusion. Numerical experiments verify the advantages of HDMI on long-horizon decision-making compared to SOTA offline RL methods and conditional generative models.
Wenhao Li 0001, Xiangfeng Wang 0001, Bo Jin 0003, Hongyuan Zha
ICML3
2023 Learning Cooperative Oversubscription for Cloud by Chance-Constrained Multi-Agent Reinforcement Learning
abstract
Oversubscription is a common practice for improving cloud resource utilization. It allows the cloud service provider to sell more resources than the physical limit, assuming not all users would fully utilize the resources simultaneously. However, how to design an oversubscription policy that improves utilization while satisfying some safety constraints remains an open problem. Existing methods and industrial practices are over-conservative, ignoring the coordination of diverse resource usage patterns and probabilistic constraints. To address these two limitations, this paper formulates the oversubscription for cloud as a chance-constrained optimization problem and proposes an effective Chance-Constrained Multi-Agent Reinforcement Learning (C2MARL) method to solve this problem. Specifically, C2MARL reduces the number of constraints by considering their upper bounds and leverages a multi-agent reinforcement learning paradigm to learn a safe and optimal coordination policy. We evaluate our C2MARL on an internal cloud platform and public cloud datasets. Experiments show that our C2MARL outperforms existing methods in improving utilization () under different levels of safety constraints.
Junjie Sheng, Lu Wang 0029, Fangkai Yang, Bo Qiao 0001, Hang Dong 0004, Xiangfeng Wang 0001, Bo Jin 0003, Jun Wang 0006, Si Qin, Saravan Rajmohan, Qingwei Lin, Dongmei Zhang 0001
WWW7
2023 F2A2: Flexible Fully-decentralized Approximate Actor-critic for Cooperative Multi-agent Reinforcement Learning
abstract
Traditional centralized multi-agent reinforcement learning (MARL) algorithms are sometimes unpractical in complicated applications due to non-interactivity between agents, the curse of dimensionality, and computation complexity. Hence, several decentralized MARL algorithms are motivated. However, existing decentralized methods only handle the fully cooperative setting where massive information needs to be transmitted in training. The block coordinate gradient descent scheme they used for successive independent actor and critic steps can simplify the calculation, but it causes serious bias. This paper proposes a flexible fully decentralized actor-critic MARL framework, which can combine most of the actor-critic methods and handle large-scale general cooperative multi-agent settings. A primal-dual hybrid gradient descent type algorithm framework is designed to learn individual agents separately for decentralization. From the perspective of each agent, policy improvement and value evaluation are jointly optimized, which can stabilize multi-agent policy learning. Furthermore, the proposed framework can achieve scalability and stability for the large-scale environment. This framework also reduces information transmission by the parameter sharing mechanism and novel modeling-other-agents methods based on theory-of-mind and online supervised learning. Sufficient experiments in cooperative Multi-agent Particle Environment and StarCraft II show that the proposed decentralized MARL instantiation algorithms perform competitively against conventional centralized and decentralized methods.
Wenhao Li 0001, Bo Jin 0003, Xiangfeng Wang 0001, Junchi Yan, Hongyuan Zha
J. Mach. Learn. Res.2
2023 Interactive medical image segmentation with self-adaptive confidence calibration
abstract
Interactive medical image segmentation based on human-in-the-loop machine learning is a novel paradigm that draws on human expert knowledge to assist medical image segmentation. However, existing methods often fall into what we call interactive misunderstanding, the essence of which is the dilemma in trading off short- and long-term interaction information. To better use the interaction information at various timescales, we propose an interactive segmentation framework, called interactive MEdical image segmentation with self-adaptive Confidence CAlibration (MECCA), which combines action-based confidence learning and multi-agent reinforcement learning. A novel confidence network is learned by predicting the alignment level of the action with short-term interaction information. A confidence-based reward-shaping mechanism is then proposed to explicitly incorporate confidence in the policy gradient calculation, thus directly correcting the model’s interactive misunderstanding. MECCA also enables user-friendly interactions by reducing the interaction intensity and difficulty via label generation and interaction guidance, respectively. Numerical experiments on different segmentation tasks show that MECCA can significantly improve short- and long-term interaction information utilization efficiency with remarkably fewer labeled samples. The demo video is available at https://bit.ly/mecca-demo-video .
Chuyun Shen, Wenhao Li 0001, Qisen Xu, Bo Jin 0003, Haibin Cai, Fengping Zhu, Xiangfeng Wang 0001
Frontiers Inf. Technol. Electron. Eng.5
2022 CLD-Net: Complement Local Detail For Medical Small-Object Segmentation
abstract
Deep learning has achieved outstanding performance in biomedical image segmentation. However, it still lacks attention to medical small-object segmentation scenarios, which is essential in early disease diagnosis. The low-resolution feature and single receptive field for medical small-object lead to a significant gap compared with the normal-scale object segmentation. This paper proposes a novel complement local detail-based network architecture (CLD-Net) for medical small-object segmentation, which can complement local detailed information when up-sampling global features. In details, the CLD-Net architecture is established with two sub-modules, i.e., Local Edge Feature Extraction Block (LEFE) and Local-Global Feature Fusion Block (LGFF). LEFE aims to maintain a high-resolution edge-feature sequence corresponding to each layer of the encoder through a progressive scheme. LGFF further rectifies the difference of features to represent the layer-aware detailed information, while the segmentation map can be obtained by incorporating the layer-aware local detailed features into the low-resolution fine-grained global feature. The experimental results on the polyp segmentation task demonstrate the effectiveness of the proposed method. CLD-Net outperforms state-of-the-art methods for small-object segmentation.
Xiangfeng Wang 0001, Bo Jin 0003, Jiaqi Tu, Fengping Zhu
BIBM3
2022 ReAssigner: A Plug-and-Play Virtual Machine Scheduling Intensifier for Heterogeneous Requests
abstract
With the rapid development of cloud computing, virtual machine scheduling has become one of the most important but challenging issues for the cloud computing community, especially for practical heterogeneous request sequences. By analyzing the impact of request heterogeneity on some popular heuristic schedulers, it can be found that existing scheduling algorithms can not handle the request heterogeneity properly and efficiently. In this paper, a plug-and-play virtual machine scheduling intensifier, called Resource Assigner (ReAssigner), is proposed to enhance the scheduling efficiency of any given scheduler for heterogeneous requests. The key idea of ReAssigner is to pre-assign roles to physical resources and let resources of the same role form a virtual cluster to handle homogeneous requests. ReAssigner can cooperate with arbitrary schedulers by restricting their scheduling space to virtual clusters. With evaluations on the real dataset from Huawei Cloud, the proposed ReAssigner achieves significant scheduling performance improvement compared with some state-of-the-art scheduling methods.
Haochuan Cui, Junjie Sheng, Bo Jin 0003, Yiqiu Hu, Xiangfeng Wang 0001
IEEE Big Data3
2022 Obtaining Dyadic Fairness by Optimal Transport
abstract
Fairness has been taken as a critical metric in machine learning models, which is considered as an important component of trustworthy machine learning. In this paper, we focus on obtaining fairness for popular link prediction tasks, which are measured by dyadic fairness. A novel pre-processing methodology is proposed to establish dyadic fairness through data repairing based on optimal transport theory. With the well-established theoretical connection between the dyadic fairness for graph link prediction and a conditional distribution alignment problem, the dyadic repairing scheme can be equivalently transformed into a conditional distribution alignment problem. Furthermore, an optimal transport-based dyadic fairness algorithm called DyadicOT is obtained by efficiently solving the alignment problem, satisfying flexibility and unambiguity requirements. The proposed DyadicOT algorithm shows superior results in obtaining fairness compared to other fairness methods on two benchmark graph datasets.
Moyi Yang, Junjie Sheng, Wenyan Liu 0001, Bo Jin 0003, Xiaoling Wang 0004, Xiangfeng Wang 0001
IEEE Big Data4
2022 Dealing with Non-Stationarity in MARL via Trust-Region Decomposition
Wenhao Li 0001, Xiangfeng Wang 0001, Bo Jin 0003, Junjie Sheng, Hongyuan Zha
ICLR3
2022 Multi-Agent Path Finding with Prioritized Communication Learning
abstract
Multi-agent pathfinding (MAPF) has been widely used to solve large-scale real-world problems, e.g., automation warehouses. The learning-based, fully decentralized framework has been introduced to alleviate real-time problems and simultaneously pursue optimal planning policy. However, existing methods might generate significantly more vertex conflicts (or collisions), which lead to a low success rate or more makespan. In this paper, we propose a PrIoritized COmmunication learning method (PICO), which incorporates the implicit planning priorities into the communication topology within the decentralized multi-agent reinforcement learning framework. Assembling with the classic coupled planners, the implicit priority learning module can be utilized to form the dynamic communication topology, which also builds an effective collision-avoiding mechanism. PICO performs significantly better in large-scale MAPF tasks in success rates and collision rates than state-of-the-art learning-based planners.
Wenhao Li 0001, Bo Jin 0003, Wenzhe Tan, Hongyuan Zha, Xiangfeng Wang 0001
ICRA3
2022 VMAgent: A Practical Virtual Machine Scheduling Platform
abstract
Virtual machine (VM) scheduling is one of the critical tasks in cloud computing. Many works have attempted to incorporate machine learning, especially reinforcement learning, to empower VM scheduling procedures. Although improved results are shown in several demo simulators, the performances in real-world scenarios are still underexploited. In this paper, we design a practical VM scheduling platform, i.e., VMAgent, to assist researchers in developing their methods on the VM scheduling problem. VMAgent consists of three components: simulator, scheduler, and visualizer. The simulator abstracts three general realistic scheduling scenarios (fading, recovering, and expansion) based on Huawei Cloud’s scheduling data, which is the core of our platform. Flexible configurations are further provided to make the simulator compatible with practical cloud computing architecture (i.e., Multi Non-Uniform Memory Access) and scenarios. Researchers then need to instantiate the scheduler to interact with the simulator, which is also pre-built in various types (e.g., heuristic, machine learning, and operations research) of scheduling algorithms to speed up the algorithm design. The visualizer, as an auxiliary component of the simulator and scheduler, facilitates researchers to conduct an in-depth analysis of the scheduling procedure and comprehensively compare different scheduling algorithms. We believe that VMAgent would shed light on the AI for the VM scheduling community, and the demo video is presented in https://bit.ly/vmagent-demo-video.
Junjie Sheng, Shengliang Cai, Haochuan Cui, Wenhao Li 0001, Yun Hua, Bo Jin 0003, Yiqiu Hu, Hongyuan Zha, Xiangfeng Wang 0001
IJCAI6
2022 Learning structured communication for multi-agent reinforcement learning
Junjie Sheng, Xiangfeng Wang 0001, Bo Jin 0003, Junchi Yan, Wenhao Li 0001, Tsung-Hui Chang, Jun Wang 0006, Hongyuan Zha
Auton. Agents Multi Agent Syst.3
2022 Mitigating disparate impact on model accuracy in differentially private learning
Wenyan Liu 0001, Xiangfeng Wang 0001, Haikun Zheng, Bo Jin 0003, Xiaoling Wang 0004, Hongyuan Zha
Inf. Sci.4
2022 Structured Cooperative Reinforcement Learning With Time-Varying Composite Action Space
abstract
In recent years, reinforcement learning has achieved excellent results in low-dimensional static action spaces such as games and simple robotics. However, the action space is usually composite, composed of multiple sub-action with different functions, and time-varying for practical tasks. The existing sub-actions might be temporarily invalid due to the external environment, while unseen sub-actions can be added to the current system. To solve the robustness and transferability problems in time-varying composite action spaces, we propose a structured cooperative reinforcement learning algorithm based on the centralized critic and decentralized actor framework, called SCORE. We model the single-agent problem with composite action space as a fully cooperative partially observable stochastic game and further employ a graph attention network to capture the dependencies between heterogeneous sub-actions. To promote tighter cooperation between the decomposed heterogeneous agents, SCORE introduces a hierarchical variational autoencoder, which maps the heterogeneous sub-action space into a common latent action space. We also incorporate an implicit credit assignment structure into the SCORE to overcome the multi-agent credit assignment problem in the fully cooperative partially observable stochastic game. Performance experiments on the proof-of-concept task and precision agriculture task show that SCORE has significant advantages in robustness and transferability for time-varying composite action space.
Wenhao Li 0001, Xiangfeng Wang 0001, Bo Jin 0003, Dijun Luo, Hongyuan Zha
IEEE Trans. Pattern Anal. Mach. Intell.3
2022 Learning to schedule multi-NUMA virtual machines via reinforcement learning
Junjie Sheng, Yiqiu Hu, Bo Jin 0003, Jun Wang 0006, Xiangfeng Wang 0001
Pattern Recognit.5
2021 Semi-supervised Medical Image Segmentation with Confidence Calibration
abstract
The lack of high-quality expert labeled data is a common shortfall for medical image segmentation, which promotes semi-supervised learning scheme to an active research topic. The pseudo-labeling technique has been demonstrated to be a powerful module in semi-supervised segmentation framework for leveraging unlabeled data. However, simple generated pseudo labels are inevitably noisy and limited by the introduced confirmation biases, for the reason that the prediction errors of these pseudo labels would enhance the misleading to the segmentation network. In this paper, we propose to estimate the prediction confidence during the training process and further utilize the confidence to calibrate the pseudo label with the purpose to mitigate the confirmation bias problem. To emphasize, the pixel-wise confidence of the prediction results are learned through an adversarial network, while untrustworthy areas could be determined based on the prediction confidence. Rectified pseudo labels on untrustworthy areas are modified and further be utilized for medical image segmentation. The effectiveness on segmentation performance and noisy pseudo label calibration are proved by comparing several supervised or semi-supervised methods on BraTS2015 dataset and other two 3D medical image datasets.
Qisen Xu, Yiqiu Hu, Bo Jin 0003, Fengping Zhu, Xiangfeng Wang 0001
IJCNN4
2021 HMRL: Hyper-Meta Learning for Sparse Reward Reinforcement Learning Problem
abstract
In spite of the success of existing meta reinforcement learning methods, they still have difficulty in learning a meta policy effectively for RL problems with sparse reward. In this respect, we develop a novel meta reinforcement learning framework called Hyper-Meta RL(HMRL), for sparse reward RL problems. It is consisted with three modules including the cross-environment meta state embedding module which constructs a common meta state space to adapt to different environments; the meta state based environment-specific meta reward shaping which effectively extends the original sparse reward trajectory by cross-environmental knowledge complementarity and as a consequence the meta policy achieves better generalization and efficiency with the shaped meta reward. Experiments with sparse-reward environments show the superiority of HMRL on both transferability and policy learning efficiency.
Yun Hua, Xiangfeng Wang 0001, Bo Jin 0003, Wenhao Li 0001, Junchi Yan, Hongyuan Zha
KDD3
2021 Distributed and Parallel ADMM for Structured Nonconvex Optimization Problem
abstract
The nonconvex optimization problems have recently attracted significant attention. However, both efficient algorithm and solid theory are still very limited. The difficulty is even pronounced for structured large-scale problems in many real-world applications. This article proposes an application-driven algorithmic framework for structured nonconvex optimization problems with distributed and parallel techniques, which jointly handles the high dimensionality of model parameters and distributed training data. The theoretical convergence of our algorithm is established under moderate assumptions. We apply the proposed method to popular multitask applications, including a multitask reinforcement learning problem. The promising performance demonstrates our framework is effective and efficient.
Xiangfeng Wang 0001, Junchi Yan, Bo Jin 0003, Wenhao Li 0001
IEEE Trans. Cybern.3
2021 Boundary-Aware Supervoxel-Level Iteratively Refined Interactive 3D Image Segmentation With Multi-Agent Reinforcement Learning
abstract
Interactive segmentation has recently been explored to effectively and efficiently harvest high-quality segmentation masks by iteratively incorporating user hints. While iterative in nature, most existing interactive segmentation methods tend to ignore the dynamics of successive interactions and take each interaction independently. We here propose to model iterative interactive image segmentation with a Markov decision process (MDP) and solve it with reinforcement learning (RL) where each voxel is treated as an agent. Considering the large exploration space for voxel-wise prediction and the dependence among neighboring voxels for the segmentation tasks, multi-agent reinforcement learning is adopted, where the voxel-level policy is shared among agents. Considering that boundary voxels are more important for segmentation, we further introduce a boundary-aware reward, which consists of a global reward in the form of relative cross-entropy gain, to update the policy in a constrained direction, and a boundary reward in the form of relative weight, to emphasize the correctness of boundary predictions. To combine the advantages of different types of interactions, i. e., simple and efficient for point-clicking, and stable and robust for scribbles, we propose a supervoxel-clicking based interaction design. Experimental results on four benchmark datasets have shown that the proposed method significantly outperforms the state-of-the-arts, with the advantage of fewer interactions, higher accuracy, and enhanced robustness.
Chaofan Ma, Qisen Xu, Xiangfeng Wang 0001, Bo Jin 0003, Xiaoyun Zhang 0001, Yanfeng Wang 0001, Ya Zhang 0002
IEEE Trans. Medical Imaging4
2020 Iteratively-Refined Interactive 3D Medical Image Segmentation With Multi-Agent Reinforcement Learning
abstract
Existing automatic 3D image segmentation methods usually fail to meet the clinic use. Many studies have explored an interactive strategy to improve the image segmentation performance by iteratively incorporating user hints. However, the dynamic process for successive interactions is largely ignored. We here propose to model the dynamic process of iterative interactive image segmentation as a Markov decision process (MDP) and solve it with reinforcement learning (RL). Unfortunately, it is intractable to use single-agent RL for voxel-wise prediction due to the large exploration space. To reduce the exploration space to a tractable size, we treat each voxel as an agent with a shared voxel-level behavior strategy so that it can be solved with multi-agent reinforcement learning. An additional advantage of this multi-agent model is to capture the dependency among voxels for segmentation task. Meanwhile, to enrich the information of previous segmentations, we reserve the prediction uncertainty in the state space of MDP and derive an adjustment action space leading to a more precise and finer segmentation. In addition, to improve the efficiency of exploration, we design a relative cross-entropy gain-based reward to update the policy in a constrained direction. Experimental results on various medical datasets have shown that our method significantly outperforms existing state-of-the-art methods, with the advantage of less interactions and a faster convergence.
Xuan Liao, Wenhao Li 0001, Qisen Xu, Xiangfeng Wang 0001, Bo Jin 0003, Xiaoyun Zhang 0001, Yanfeng Wang 0001, Ya Zhang 0002
CVPR5
2020 VSB2-Net: Visual-Semantic Bi-Branch Network for Zero-Shot Hashing
abstract
Zero-shot hashing aims at learning hashing model from seen classes and the obtained model is capable of generalizing to unseen classes for image retrieval. Inspired by zero-shot learning, existing zero-shot hashing methods usually transfer the supervised knowledge from seen to unseen classes, by embedding the hamming space to a shared semantic space. However, this makes instances difficult to distinguish due to limited hashing bit numbers, especially for semantically similar unseen classes. We propose a novel inductive zero-shot hashing framework, i.e., VSB2-Net, where both semantic space and visual feature space are embedded to the same hamming space instead. The reconstructive semantic relationships are established in the hamming space, preserving local similarity relationships and explicitly enlarging the discrepancy between semantic hamming vectors. A two-task architecture, comprising of classification module and visual feature reconstruction module, is employed to enhance the generalization and transfer abilities. Extensive evaluation results on several benchmark datasets demonstrate the superiority of our proposed method compared to several state-of-the-art baselines.
Xiangfeng Wang 0001, Bo Jin 0003, Jun Wang 0006, Hongyuan Zha
ICPR3
2020 Heterogeneous Graph-based Knowledge Transfer for Generalized Zero-shot Learning
abstract
Generalized zero-shot learning (GZSL) tackles the problem of learning to classify instances involving both seen classes and unseen ones. The key issue is how to effectively transfer the model learned from seen classes to unseen classes. Existing works in GZSL usually assume that some prior information about unseen classes are available. However, such an assumption is unrealistic when new unseen classes appear dynamically. To this end, we propose a novel heterogeneous graph-based knowledge transfer method (HGKT) for GZSL, agnostic to unseen classes and instances, by leveraging graph neural network. Specifically, a structured heterogeneous graph is constructed with high-level representative nodes for seen classes, which are chosen through Wasserstein barycenter in order to simultaneously capture inter-class and intra-class relationship. The aggregation and embedding functions can be learned through graph neural network, which can be used to compute the embeddings of unseen classes by transferring the knowledge from their neighbors. Extensive experiments on public benchmark datasets show that our method achieves state-of-the-art results.
Xiangfeng Wang 0001, Bo Jin 0003, Junchi Yan, Hongyuan Zha
ICPR3
2020 Visual-to-Semantic Hashing for Zero Shot Learning
abstract
Hashing-based multimedia retrieval are facing the problem of the dramatic increase of data, especially new unseen categories. It is time-consuming, expensive, and sometimes impractical to label new samples and retrain the hashing model. Recently, several zero-shot hashing methods are proposed to generate the hash function with good generalization for unseen classes, via exploring semantic information and similarity relationship. However, the performance of existing solutions is still not satisfying. Therefore, we propose a modified two-stage framework, called Visual-to-Semantic Hashing (VSH). To fully exploit the semantic information, visual feature is firstly mapped to the semantic space, and then generate its hash codes. To transfer supervised knowledge from seen classes to unseen classes, a margin-based ranking loss is employed to learn the semantic structure. To boost the discriminability of semantic mapping, a classification module is adopted to distinguish between different semantic mapping vectors. Plenty of experiments demonstrate that the proposed VSH is superior to state-of-the-art methods.
Xiaoyue Wen, Bo Jin 0003, Xiangfeng Wang 0001, Jinghui Cai
IJCNN3
2018 Learning to Detect Pathogenic Microorganism of Community-acquired Pneumonia
abstract
Community-acquired pneumonia (CAP) is a major death cause for children, requiring an early administration of appropriate antibiotics to cure it. To achieve this, accurate detection of pathogenic microorganism is crucial, especially for reducing the abuse of antibiotics. Conventional gold standard detection methods are mainly etiology based, incurring high cost and labor intensity. Although recently electronic health records (EHRs) become prevalent and widely used, their power for automatically determining pathogenic microorganism has not been investigated. In this paper, we formulate a new problem for automatically detecting pathogenic microorganism of CAP by considering patient biomedical features from EHRs, including time-varying body temperatures and common laboratory measurements. We further develop a Patient Attention based Recurrent Neural Network (PA-RNN) model to fuse different patient features for detection. We conduct experiments on a real dataset, demonstrating utilizing electronic health records yields promising performance and PA-RNN outperforms several alternatives.
Wenwei Liang, Wei Zhang 0056, Bo Jin 0003, Jiangjiang Xu, Linhua Shu, Hongyuan Zha
SIGIR3
2016 On Modeling and Predicting Individual Paper Citation Count over Time
Shuai Xiao 0002, Junchi Yan, Bo Jin 0003, Xiangfeng Wang 0001, Xiaokang Yang 0001, Stephen M. Chu, Hongyuan Zha
IJCAI4
2016 Modeling Contagious Merger and Acquisition via Point Processes with a Profile Regression Prior
Junchi Yan, Shuai Xiao 0002, Bo Jin 0003, Xiangfeng Wang 0001, Bin Ke, Xiaokang Yang 0001, Hongyuan Zha
IJCAI4
2016 A compressive tracking based on time-space Kalman fusion model
Xiao Yun, Zhongliang Jing, Bo Jin 0003, Canlong Zhang
Sci. China Inf. Sci.4
2013 Locally discriminative stable model for visual tracking with clustering and principle component analysis
abstract
The challenge of visual tracking mainly comes from intrinsic appearance variations of the target and extrinsic environment changes around the target in a long duration, so the tracker that can simultaneously tolerate these variabilities is largely expected. In this study, the authors propose a new tracking approach based on discriminative stable regions (DSRs). The DSRs are obtained based on the criterion of maximal local entropy and spatial discrimination, which enables the tracker to handle well distractors and appearance variations. The collaborative tracking incorporated hierarchical clustering can tolerate motion noise and occlusions. In addition, as an efficient tool, the principle component analysis is used to discover the potential affine relation between DSR and the target, which timely adapts to the shape deformation of the target. Extensive experiments show that the proposed method achieves superior performance in many challenging target tracking tasks.
Canlong Zhang, Zhongliang Jing, Yanping Tang, Bo Jin 0003
IET Comput. Vis.4
2013 A sparse proximal Newton splitting method for constrained image deblurring
Han Pan, Zhongliang Jing, Rongli Liu, Bo Jin 0003, Canlong Zhang
Neurocomputing5
2013 Robust visual tracking using discriminative stable regions and K-means clustering
Canlong Zhang, Zhongliang Jing, Han Pan, Bo Jin 0003, Zhixin Li 0001
Neurocomputing4
2012 A dual-kernel-based tracking approach for visual target
Canlong Zhang, Zhongliang Jing, Bo Jin 0003, Zhixin Li 0001
Sci. China Inf. Sci.3