Demo Data from SIGIR 2026 / reviewers in the wild / expert
Jun Wang 0012
dblp:w/JunWang12
· DBLP profile ↗
258ranked-venue papers
20as first author
142since 2021 · last 2026
—ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 167 · 5 first-author · 108 since 2021Databases, data management, data science and information retrieval · 106 · 15 first-author · 26 since 2021Graphics, computer vision, multimedia, augmented reality and games · 38 · 4 first-author · 16 since 2021Applied, interdisciplinary, general and emerging computing · 12 · 8 since 2021Computer networks · 9 · 8 since 2021Systems, architecture and hardware · 8 · 7 since 2021Software engineering, systems software and programming languages · 1 · 1 since 2021Human-computer interaction and ubiquitous computing · 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
115 papers |
Reinforcement learning · 40% Language models and text generation · 15% Efficient and distributed learning · 9% | |
| Databases, data mining, and information retrieval
46 papers |
Information retrieval · 55% Recommender systems · 35% Data mining · 5% | |
| Theoretical computer science
21 papers |
Algorithmic game theory and mechanism design · 58% Mathematical optimization · 34% Algorithms and data structures · 8% | |
| Interdisciplinary, comprehensive, and emerging computing
11 papers |
Computational finance and economics · 37% Computational social science and digital humanities · 32% Computational science and engineering · 22% |
Topics — the 30 heaviest of 365, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Reinforcement learning
multi-agent reinforcement learning |
13.3 | 26 | 2025 | ReMA: Learning to Meta-Think for LLMs with Multi-agent Reinforcement Learning · NeurIPS 2025 A Principle of Targeted Intervention for Multi-Agent Reinforcement Learning · NeurIPS 2025 MALib: A Parallel Framework for Population-based Multi-agent Reinforcement Learning · J. Mach. Learn. Res. 2023 |
Machine learning › Reinforcement learning
safe reinforcement learning |
4.3 | 6 | 2026 | Proactive Constrained Policy Optimization with Preemptive Penalty · AAAI 2026 Fear-Neuro-Inspired Reinforcement Learning for Safe Autonomous Driving · IEEE Trans. Pattern Anal. Mach. Intell. 2024 A Review of Safe Reinforcement Learning: Methods, Theories, and Applications · IEEE Trans. Pattern Anal. Mach. Intell. 2024 |
Machine learning › Reinforcement learning
policy optimization |
4.1 | 6 | 2026 | Proactive Constrained Policy Optimization with Preemptive Penalty · AAAI 2026 Uncertainty-quantified Rollout Policy Adaptation for Unlabelled Cross-domain Video Temporal Grounding · NeurIPS 2025 Reinforcing LLM Agents via Policy Optimization with Action Decomposition · NeurIPS 2024 |
Natural language and speech › Language models and text generation
LLM agents |
2.6 | 4 | 2025 | MF-LLM: Simulating Population Decision Dynamics via a Mean-Field Large Language Model Framework · NeurIPS 2025 Reinforcing LLM Agents via Policy Optimization with Action Decomposition · NeurIPS 2024 Large Language Models Are Neurosymbolic Reasoners · AAAI 2024 |
Natural language and speech › Language models and text generation
chain-of-thought reasoning |
2.6 | 3 | 2025 | Self-Verifying Reflection Helps Transformers with CoT Reasoning · NeurIPS 2025 Causal Sufficiency and Necessity Improves Chain-of-Thought Reasoning · NeurIPS 2025 Direct Value Optimization: Improving Chain-of-Thought Reasoning in LLMs with Refined Values · EMNLP 2025 |
Natural language and speech › Language models and text generation
alignment |
2.5 | 3 | 2025 | Risk-aware Direct Preference Optimization under Nested Risk Measure · NeurIPS 2025 Constrain Alignment with Sparse Autoencoders · ICML 2025 Token-level Direct Preference Optimization · ICML 2024 |
Natural language and speech › Language models and text generation
large language model reasoning |
2.5 | 3 | 2025 | ReMA: Learning to Meta-Think for LLMs with Multi-agent Reinforcement Learning · NeurIPS 2025 ThinkBench: Dynamic Out-of-Distribution Evaluation for Robust LLM Reasoning · NeurIPS 2025 Policy Learning from Tutorial Books via Understanding, Rehearsing and Introspecting · NeurIPS 2024 |
Machine learning › Reinforcement learning › policy optimization
policy gradient |
2.4 | 6 | 2022 | A Theoretical Understanding of Gradient Bias in Meta-Reinforcement Learning · NeurIPS 2022 Understanding Policy Gradient Algorithms: A Sensitivity-Based Approach · ICML 2022 Settling the Variance of Multi-Agent Policy Gradients · NeurIPS 2021 |
Machine learning › Reinforcement learning › multi-agent reinforcement learning
credit assignment |
2.0 | 3 | 2024 | Reinforcing LLM Agents via Policy Optimization with Action Decomposition · NeurIPS 2024 Interpretable Reward Redistribution in Reinforcement Learning: A Causal Approach · NeurIPS 2023 Multiagent Q-learning with Sub-Team Coordination · NeurIPS 2022 |
Knowledge, reasoning and agents › Knowledge representation and reasoning
causal reasoning |
1.9 | 3 | 2025 | Curious Causality-Seeking Agents in Open-ended Worlds · NeurIPS 2025 Causal Sufficiency and Necessity Improves Chain-of-Thought Reasoning · NeurIPS 2025 CausalVAE: Disentangled Representation Learning via Neural Structural Causal Models · CVPR 2021 |
Machine learning › Efficient and distributed learning › inference acceleration
speculative decoding |
1.9 | 2 | 2026 | Probe-and-Fetch: Dynamic KV Cache Pruning for Accelerated Long-Context Inference in Web-Scale AI Search · WWW 2026 Mixture of Attentions For Speculative Decoding · ICLR 2025 |
Natural language and speech › Language models and text generation › preference optimization
direct preference optimization |
1.6 | 2 | 2025 | Risk-aware Direct Preference Optimization under Nested Risk Measure · NeurIPS 2025 Token-level Direct Preference Optimization · ICML 2024 |
Natural language and speech › Language models and text generation
preference optimization |
1.6 | 2 | 2025 | Constrain Alignment with Sparse Autoencoders · ICML 2025 Token-level Direct Preference Optimization · ICML 2024 |
Program synthesis and code generation
code generation with language models |
1.6 | 2 | 2025 | Optimizing Case-Based Reasoning System for Functional Test Script Generation with Large Language Models · KDD (2) 2025 DS-Agent: Automated Data Science by Empowering Large Language Models with Case-Based Reasoning · ICML 2024 |
Machine learning › Probabilistic and Bayesian machine learning
causal inference |
1.5 | 2 | 2025 | A Principle of Targeted Intervention for Multi-Agent Reinforcement Learning · NeurIPS 2025 Invariant Learning via Probability of Sufficient and Necessary Causes · NeurIPS 2023 |
Machine learning › Reinforcement learning › large-scale reinforcement learning
distributed reinforcement learning |
1.5 | 2 | 2025 | DistRL: An Asynchronous Distributed Reinforcement Learning Framework for On-Device Control Agent · ICLR 2025 GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models · ICML 2023 |
Machine learning › Reinforcement learning › off-policy reinforcement learning
experience replay |
1.5 | 2 | 2025 | Succeed or Learn Slowly: Sample Efficient Off-Policy Reinforcement Learning for Mobile App Control · NeurIPS 2025 GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models · ICML 2023 |
Recommender systems
collaborative filtering |
1.5 | 8 | 2023 | CounterCLR: Counterfactual Contrastive Learning with Non-random Missing Data in Recommendation · ICDM 2023 Probabilistic group recommendation via information matching · WWW 2013 Optimizing top-n collaborative filtering via dynamic negative item sampling · SIGIR 2013 |
Machine learning › Reinforcement learning
policy learning |
1.4 | 2 | 2024 | Policy Learning from Tutorial Books via Understanding, Rehearsing and Introspecting · NeurIPS 2024 ChessGPT: Bridging Policy Learning and Language Modeling · NeurIPS 2023 |
Machine learning › Efficient and distributed learning
model compression |
1.4 | 2 | 2026 | Dual Activation-Weight Sparsity: A Training-Free Framework for Efficient Large Language Model Compression · ACL (1) 2026 BayesNAS: A Bayesian Approach for Neural Architecture Search · ICML 2019 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference
causal discovery |
1.4 | 2 | 2025 | Curious Causality-Seeking Agents in Open-ended Worlds · NeurIPS 2025 Ordering-Based Causal Discovery with Reinforcement Learning · IJCAI 2021 |
Machine learning › Reinforcement learning › multi-agent reinforcement learning
opponent modeling |
1.4 | 4 | 2020 | Modelling Bounded Rationality in Multi-Agent Interactions by Generalized Recursive Reasoning · IJCAI 2020 Learning to Model Opponent Learning (Student Abstract) · AAAI 2020 A Regularized Opponent Model with Maximum Entropy Objective · IJCAI 2019 |
Information retrieval
ranking |
1.4 | 3 | 2024 | InfoRank: Unbiased Learning-to-Rank via Conditional Mutual Information Minimization · WWW 2024 A Deep Recurrent Survival Model for Unbiased Ranking · SIGIR 2020 Interactive exploratory search for multi page search results · WWW 2013 |
Information retrieval › ranking
learning to rank |
1.3 | 2 | 2024 | InfoRank: Unbiased Learning-to-Rank via Conditional Mutual Information Minimization · WWW 2024 Beyond Relevance Ranking: A General Graph Matching Framework for Utility-Oriented Learning to Rank · ACM Trans. Inf. Syst. 2022 |
Information retrieval › ranking › learning to rank
unbiased learning to rank |
1.3 | 2 | 2024 | InfoRank: Unbiased Learning-to-Rank via Conditional Mutual Information Minimization · WWW 2024 Beyond Relevance Ranking: A General Graph Matching Framework for Utility-Oriented Learning to Rank · ACM Trans. Inf. Syst. 2022 |
Machine learning › Efficient and distributed learning
distributed training |
1.3 | 2 | 2023 | MALib: A Parallel Framework for Population-based Multi-agent Reinforcement Learning · J. Mach. Learn. Res. 2023 GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models · ICML 2023 |
Robotics › Legged, aerial and field robots › legged robots
legged robot locomotion |
1.3 | 2 | 2023 | Multi-embodiment Legged Robot Control as a Sequence Modeling Problem · ICRA 2023 Sim-to-Real Transfer for Quadrupedal Locomotion via Terrain Transformer · ICRA 2023 |
Recommender systems
debiased recommendation |
1.3 | 2 | 2023 | Debiased Recommendation with User Feature Balancing · ACM Trans. Inf. Syst. 2023 CounterCLR: Counterfactual Contrastive Learning with Non-random Missing Data in Recommendation · ICDM 2023 |
Machine learning › Reinforcement learning
off-policy reinforcement learning |
1.3 | 2 | 2025 | Succeed or Learn Slowly: Sample Efficient Off-Policy Reinforcement Learning for Mobile App Control · NeurIPS 2025 Off-Policy Reinforcement Learning for Efficient and Effective GAN Architecture Search · ECCV (7) 2020 |
Machine learning › Reinforcement learning › reward design
reward shaping |
1.2 | 2 | 2023 | Learning to Shape Rewards Using a Game of Two Partners · AAAI 2023 LIGS: Learnable Intrinsic-Reward Generation Selection for Multi-Agent Learning · ICLR 2022 |
Methods — techniques the papers use, named apart from their topics
reinforcement learning · 10.1large language model · 6.4case-based reasoning · 4.0actor-critic · 3.5policy gradient · 2.4variational autoencoder · 1.8transformer · 1.7markov decision process · 1.7group relative policy optimization · 1.7direct preference optimization · 1.6bradley-terry model · 1.6KL divergence · 1.6multi-agent reinforcement learning · 1.4online PCA · 1.3fixed-point iteration · 1.3policy-space response oracle · 1.2discrete optimization · 1.0speculative decoding · 1.0
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Vision-Language Reasoning for Geolocalization: A Reinforcement Learning ApproachabstractRecent advances in vision-language models have opened up new possibilities for reasoning-driven image geolocalization. However, existing approaches often rely on synthetic reasoning annotations or external image retrieval, which can limit interpretability and generalizability. In this paper, we present Geo-R, a retrieval-free framework that uncovers structured reasoning paths from existing ground-truth coordinates and optimizes geolocation accuracy via reinforcement learning. We propose the Chain of Region, a rule-based hierarchical reasoning paradigm that generates precise, interpretable supervision by mapping GPS coordinates to geographic entities (e.g., country, province, city) without relying on model-generated or synthetic labels. Building on this, we introduce a lightweight reinforcement learning strategy with coordinate-aligned rewards based on Haversine distance, enabling the model to refine predictions through spatially meaningful feedback. Our approach bridges structured geographic reasoning with direct spatial supervision, yielding improved localization accuracy, stronger generalization, and more transparent inference. Experimental results across multiple benchmarks confirm the effectiveness of Geo-R, establishing a new retrieval-free paradigm for scalable and interpretable image geolocalization. To facilitate further research and ensure reproducibility, both the model and code will be made publicly available. Biao Wu 0006, Ling Chen 0006, Tao Cheng 0004, Jun Wang 0012 |
AAAI | 6 |
| 2026 | Proactive Constrained Policy Optimization with Preemptive PenaltyabstractSafe Reinforcement Learning (RL) often faces significant issues such as constraint violations and instability, necessitating the use of constrained policy optimization, which seeks optimal policies while ensuring adherence to specific constraints like safety. Typically, constrained optimization problems are addressed by the Lagrangian method, a post-violation remedial approach that may result in oscillations and overshoots. Motivated by this, we propose a novel method named Proactive Constrained Policy Optimization (PCPO) that incorporates a preemptive penalty mechanism. This mechanism integrates barrier items into the objective function as the policy nears the boundary, imposing a cost. Meanwhile, we introduce a constraint-aware intrinsic reward to guide boundary-aware exploration, which is activated only when the policy approaches the constraint boundary. We establish theoretical upper and lower bounds for the duality gap and the performance of the PCPO update, shedding light on the method's convergence characteristics. Additionally, to enhance the optimization performance, we adopt a policy iteration approach. An interesting finding is that PCPO demonstrates significant stability in experiments. Experimental results indicate that the PCPO framework provides a robust solution for policy optimization under constraints, with important implications for future research and practical applications. Ning Yang 0005, Haifeng Zhang 0002, Jun Wang 0012 |
AAAI | 6 |
| 2026 | Dual Activation-Weight Sparsity: A Training-Free Framework for Efficient Large Language Model CompressionabstractLuoyang Sun, Guangyan Li, Cheng Deng, Haifeng Zhang, Jian Zhao, Yongqiang Tang, Wensheng Zhang, Jun Wang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Luoyang Sun, Guangyan Li, Cheng Deng 0001, Haifeng Zhang 0002, Jian Zhao 0006, Yongqiang Tang, Wensheng Zhang 0002, Jun Wang 0012 |
ACL (1) | 8 |
| 2026 | Towards Unified Affective Reasoning via In-Context Reinforcement LearningabstractRecent advances in Large Language Models (LLMs) have shifted affective computing from direct label prediction to deliberative reasoning, with Reinforcement Learning (RL) emerging as a promising means of cultivating such capabilities. However, existing RL-based methods are largely confined to single-task settings, requiring separate training pipelines for different affective domains and limiting transfer across related problems. To address this limitation, we propose Unified Affective Reasoning (UAR), an in-context reinforcement learning framework for learning generalizable affective reasoning under a unified multi-task paradigm. UAR jointly optimizes multiple affective tasks and leverages in-context demonstrations to reconcile heterogeneous annotation schemas across datasets, thereby aligning disparate label spaces while preserving task-specific supervision. Experiments on 16 standard benchmarks across emotion recognition, aspect-based sentiment analysis, stance detection, and sarcasm detection show that UAR consistently improves the underlying base models and generalizes effectively to held-out out-of-domain tasks. These results demonstrate that in-context reinforcement learning provides an effective path toward unified and generalizable affective reasoning. Code is available at https://github.com/farisxiong/UAR. Jun Wang 0012, Ruifeng Xu 0001 |
SIGIR | 2 |
| 2026 | Probe-and-Fetch: Dynamic KV Cache Pruning for Accelerated Long-Context Inference in Web-Scale AI SearchabstractGenerative inference with Large Language Models (LLMs) is the cornerstone of web-scale AI search, where queries are answered using vast, heterogeneous documents retrieved via Retrieval-Augmented Generation (RAG). This paradigm is critically bottlenecked by the cost of self-attention mechanism on long context. The sheer diversity of retrieved web content (multi-sourced, multi-lingual, multi-faceted) makes simple Key-Value (KV) cache optimizations with pre-fixed subsets ineffective, demanding a dynamic, content-aware approach. This challenge, however, introduces a classic chicken-and-egg problem: the model cannot foresee the necessary KV entries for attention without first inferring on the content, yet doing so on the full context is prohibitively expensive. This paper introduces P&F, a unified framework that resolves this dilemma through a core ''probe-and-fetch'' mechanism, which ingeniously integrates with speculative decoding -- an acceleration approach already adopted in web-scale AI search. The probe step repurposes the speculative draft model: while generating candidate tokens, it simultaneously probes the context to predict the most salient KV entries the large model will need for attention. The fetch step immediately acts on this prediction, asynchronously fetching these sparse entries from memory. This synergistic design piggybacks the probing step onto the drafting process, allowing the expensive gathering of a sparse KV cache to be fully masked. Crucially, this co-design breaks the sequential dependency bottleneck that cripples naive integrations of speculative decoding and prefetching due to synchronization issues. Extensive experiments show P&F significantly outperforms state-of-the-art methods in throughput and scalability, offering a practical, drop-in solution. Extensive offline evaluations across various settings and datasets demonstrate that P&F yields superior throughput and scalability compared to advanced baselines, while maintaining model quality across diverse models and scales. In online settings, P&F delivers substantial gains in throughput improvements while preserving response quality, making it well-suited for large-scale industrial deployment in real-time AI Search services. Yuchen Li 0006, Chengzhe Zhang, Cheng Deng 0001, Xinyu Ma 0001, Tianhao Peng 0002, Hengyi Cai, Shuaiqiang Wang, Jiashu Zhao, Haoyi Xiong, Jimmy Huang 0001, Lei Chen 0002, Jun Wang 0012, Dawei Yin 0001 |
WWW | 17 |
| 2026 | Privacy Preserved Blood Glucose Level Cross-Prediction: An Asynchronous Decentralized Federated Learning ApproachabstractNewly diagnosed Type 1 Diabetes (T1D) patients often struggle to obtain effective Blood Glucose (BG) prediction models due to the lack of sufficient BG data from Continuous Glucose Monitoring (CGM), presenting a significant "cold start" problem in patient care. Utilizing population models to address this challenge is a potential solution, but collecting patient data for training population models in a privacy-conscious manner is challenging, especially given that such data is often stored on personal devices. Considering the privacy protection and addressing the "cold start" problem in diabetes care, we propose "GluADFL", blood Glucose prediction by Asynchronous Decentralized Federated Learning. We compared GluADFL with eight baseline methods using four distinct T1D datasets, comprising 298 participants, which demonstrated its superior performance in accurately predicting BG levels for cross-patient analysis. Furthermore, patients' data might be stored and shared across various communication networks in GluADFL, ranging from highly interconnected (e.g., random, performs the best among others) to more structured topologies (e.g., cluster and ring), suitable for various social networks. The asynchronous training framework supports flexible participation. By adjusting the ratios of inactive participants, we found it remains stable if less than 70% are inactive. Our results confirm that GluADFL offers a practical, privacy-preserved solution for BG prediction in T1D, significantly enhancing the quality of diabetes management. Chengzhe Piao, Taiyu Zhu, Yu Wang 0115, Stephanie E. Baldeweg, Pantelis Georgiou, Jun Wang 0012, Kezhi Li |
IEEE J. Biomed. Health Informatics | 8 |
| 2025 | MultiNFT: A Multimodal Dataset for Non-Fungible Tokens Market AnalysisabstractNon-fungible tokens (NFTs) are unique digital assets that play an increasingly important role in decentralized markets, supporting new forms of ownership, valuation, and exchange. Their inherently multimodal structure, which encompasses visual content, metadata, and trading history, has led to a growing academic interest in modeling NFT pricing and market behavior. However, existing research is limited by the lack of comprehensive datasets that unify these modalities with consistent formatting and longitudinal coverage. To address this gap, we introduce MultiNFT, a large-scale multimodal dataset comprising 50 curated profile picture (PFP) NFT collections, including 523,020 unique assets and 2.38 million transaction records from April 2021 to September 2025. MultiNFT integrates standardized images, structured metadata, and time-series trading data, along with rarity scores and aesthetic features, offering a unified foundation for multimodal learning and NFT analytics. Unlike prior datasets that focus on visual similarity or static snapshots, MultiNFT captures evolving valuation dynamics across market cycles and connects them to trait-level characteristics. We demonstrate the utility of the dataset through three case studies, including within-collection rarity-price analysis, visual feature clustering across collections, and quantifying feature contributions in a comprehensive pricing model. By bridging computer vision, behavioral modeling, and financial forecasting, MultiNFT supports a wide range of interdisciplinary research and practical use cases. The dataset is publicly available and is intended to promote reproducible experimentation and further exploration of the mechanisms driving value in digital asset ecosystems. Shuying Liu, Bowei Chen 0001, Cathy Yi-Hsuan Chen, Jun Wang 0012 |
IEEE Big Data | 5 |
| 2025 | PIE: Permutation-Invariant Multi-Entity EvaluationabstractThe evaluation system is devised in online competitive games and sports to assess players' skills. Popular methods Elo, designed for two-player competitive games such as chess and tennis, are based on the Bradley-Terry model and update players' ratings with competition outcomes. Extended methods Trueskill and mElo are proposed for multi-player(team) and two-player intransitive games, respectively. However, existing evaluation methods are constrained in specific situations; for example, mElo is limited to dealing with two-player games, and TrueSkill, Elo can not handle intransitive games. In addition, previous team evaluation methods bake in the assumption that individuals' contributions to team performance are uniformly determined by individual abilities, which does not hold in many situations, such as football with different roles and games with team score being defined as the maximum or minimum of players' gains. In this paper, we address the challenge of evaluating player skill in multi-player (team) competitions. We propose PIE, an online permutation-invariant evaluation model for multi-entity competitions that ensures the predicted winner of a multi-entity match remains invariant to the order of input entities. For multi-team evaluation, PIE enables team ratings to increase monotonically with improvements in individual player ratings. Empirical results of predicting the winner and winning probabilities in real-world games demonstrate that PIE achieves comparable performance in handling the prediction of multiplayer(team) matches with other baselines. Haifeng Zhang 0002, Yali Du 0001, Jun Wang 0012 |
CoG | 4 |
| 2025 | Learning Macroeconomic Policies Through Dynamic Stackelberg Mean-Field GamesabstractMacroeconomic outcomes emerge from individuals’ decisions, making it essential to model how agents interact with macro policy via consumption, investment, and labor choices. We formulate this as a dynamic Stackelberg game: the government (leader) sets policies, and agents (followers) respond by optimizing their behavior over time. Unlike static models, this dynamic formulation captures temporal dependencies and strategic feedback critical to policy design. However, as the number of agents increases, explicitly simulating all agent–agent and agent–government interactions becomes computationally infeasible. To address this, we propose the Dynamic Stackelberg Mean Field Game (DSMFG) framework, which approximates these complex interactions via agent–population and government–population couplings. This approximation preserves individual-level feedback while ensuring scalability, enabling DSMFG to jointly model three core features of real-world policy-making: dynamic feedback, asymmetry, and large-scale. We further introduce Stackelberg Mean Field Reinforcement Learning (SMFRL), a data-driven algorithm that learns the leader’s optimal policies while maintaining personalized responses for individual agents. Empirically, we validate our approach in a large-scale simulated economy, where it scales to 1,000 agents (vs. 100 in prior work) and achieves a 4× GDP gain over classical economic methods and a 19× improvement over the static 2022 U.S. federal income tax policy. Qirui Mi, Chengdong Ma, Si-Yu Xia, Yan Song 0003, Mengyue Yang, Jun Wang 0012, Haifeng Zhang 0002 |
ECAI | 7 |
| 2025 | Direct Value Optimization: Improving Chain-of-Thought Reasoning in LLMs with Refined ValuesabstractWe introduce Direct Value Optimization (DVO), an innovative reinforcement learning framework for enhancing large language models in complex reasoning tasks.Unlike traditional methods relying on preference labels, DVO utilizes value signals at individual reasoning steps, optimizing models via a mean squared error loss.The key benefit of DVO lies in its fine-grained supervision, circumventing the need for labor-intensive human annotations.Target values within the DVO are estimated using either Monte Carlo Tree Search or an outcome value model.Our empirical analysis on both mathematical and commonsense reasoning tasks shows that DVO consistently outperforms existing offline preference optimization techniques, even with fewer training steps.These findings underscore the importance of value signals in advancing reasoning capabilities and highlight DVO as a superior methodology under scenarios lacking explicit human preference information.The code is available at https://github.com/StevenZHB/DVO. Guangsheng Bao, Linyi Yang, Jun Wang 0012, Yue Zhang 0004 |
EMNLP | 5 |
| 2025 | Variational Perturbation Personalized Federated Learning via Prior-Posterior DistanceabstractPersonalized Federated Learning (pFL) mitigates the impact of statistical heterogeneity on FL architecture to some extent by allowing participants to use personalized models based on local data distributions. The existing pFL methods optimize from the perspective of model structure, attempting to adopt strategies that maintain model processing or quickly adapt to local data distribution capabilities. Our proposed method draws inspiration from the concept of variational inference, guiding model updates by comparing prior and posterior data distributions, and innovatively applying model variational perturbations to improve robustness. Finally, we conducted multidimensional experiments and the results show that our method outperforms the current baseline. Code: https://github.com/RezinChow/VPFL. Hefeng Zhou, Jun Wang 0012, Jiong Lou, Wugedele Bao, Chentao Wu, Jie Li 0002 |
ICASSP | 3 |
| 2025 | Circuit Transformer: A Transformer That Preserves Logical EquivalenceabstractImplementing Boolean functions with circuits consisting of logic gates is fundamental in digital computer design. However, the implemented circuit must be exactly equivalent, which hinders generative neural approaches on this task due to their occasionally wrong predictions. In this study, we introduce a generative neural model, the “Circuit Transformer”, which eliminates such wrong predictions and produces logic circuits strictly equivalent to given Boolean functions. The main idea is a carefully designed decoding mechanism that builds a circuit step-by-step by generating tokens, which has beneficial “cutoff properties” that block a candidate token once it invalidate equivalence. In such a way, the proposed model works similar to typical LLMs while logical equivalence is strictly preserved. A Markov decision process formulation is also proposed for optimizing certain objectives of circuits. Experimentally, we trained an 88-million-parameter Circuit Transformer to generate equivalent yet more compact forms of input circuits, outperforming existing neural approaches on both synthetic and real world benchmarks, without any violation of equivalence constraints.
Code: https://github.com/snowkylin/circuit-transformer Xihan Li 0001, Xing Li 0023, Lei Chen 0002, Mingxuan Yuan, Jun Wang 0012 |
ICLR | 6 |
| 2025 | Spa-Bench: a comprehensive Benchmark for Smartphone Agent EvaluationabstractSmartphone agents are increasingly important for helping users control devices efficiently, with (Multimodal) Large Language Model (MLLM)-based approaches emerging as key contenders. Fairly comparing these agents is essential but challenging, requiring a varied task scope, the integration of agents with different implementations, and a generalisable evaluation pipeline to assess their strengths and weaknesses. In this paper, we present SPA-Bench, a comprehensive SmartPhone Agent Benchmark designed to evaluate (M)LLM-based agents in an interactive environment that simulates real-world conditions. SPA-Bench offers three key contributions: (1) A diverse set of tasks covering system and third-party apps in both English and Chinese, focusing on features commonly used in daily routines; (2) A plug-and-play framework enabling real-time agent interaction with Android devices, integrating over ten agents with the flexibility to add more; (3) A novel evaluation pipeline that automatically assesses agent performance across multiple dimensions, encompassing seven metrics related to task completion and resource consumption. Our extensive experiments across tasks and agents reveal challenges like interpreting mobile user interfaces, action grounding, memory retention, and execution costs. We propose future research directions to ease these difficulties, moving closer to real-world smartphone agent applications. Jingxuan Chen, Derek Yuen, Yuhao Yang 0008, Gongwei Chen, Li Yixing, Xurui Zhou, Weiwen Liu, Shuai Wang 0020, Kaiwen Zhou 0001, Rui Shao 0001, Liqiang Nie, Yasheng Wang, Jianye Hao, Jun Wang 0012, Kun Shao |
ICLR | 16 |
| 2025 | Lightweight Neural App ControlabstractThis paper introduces a novel mobile phone control architecture, Lightweight Multi-modal App Control (LiMAC), for efficient interactions and control across various Android apps. LiMAC takes as input a textual goal and a sequence of past mobile observations, such as screenshots and corresponding UI trees, to generate precise actions. To address the computational constraints inherent to smartphones, we introduce a small Action Transformer (AcT) integrated with a fine-tuned vision-language model (VLM) for real-time decision-making and task execution. We evaluate LiMAC on two open-source mobile control datasets, demonstrating the superior performance of our small-form-factor approach against fine-tuned versions of open-source VLMs, such as Florence2 and Qwen2-VL. It also significantly outperforms prompt engineering baselines utilising closed-source foundation models like GPT-4o. More specifically, LiMAC increases the overall action accuracy by up to 19% compared to fine-tuned VLMs, and up to 42% compared to prompt-engineering baselines. Filippos Christianos, Georgios Papoudakis, Thomas Coste, Jianye Hao, Jun Wang 0012, Kun Shao |
ICLR | 5 |
| 2025 | Human-inspired Episodic Memory for Infinite Context LLMsabstractLarge language models (LLMs) have shown remarkable capabilities, but still struggle with processing extensive contexts, limiting their ability to maintain coherence and accuracy over long sequences. In contrast, the human brain excels at organising and retrieving episodic experiences across vast temporal scales, spanning a lifetime. In this work, we introduce EM-LLM, a novel approach that integrates key aspects of human episodic memory and event cognition into LLMs with no fine-tuning, enabling them to handle practically infinite context lengths while maintaining computational efficiency. EM-LLM organises sequences of tokens into coherent episodic events using a combination of Bayesian surprise and graph-theoretic boundary refinement in an online fashion. When needed, these events are retrieved through a two-stage memory process, combining similarity-based and temporally contiguous retrieval for efficient, human-inspired access to relevant information. Experiments on the LongBench and $\infty$-Bench benchmarks demonstrate EM-LLM's superior performance, consistently outperforming the state-of-the-art retrieval model InfLLM across various baseline LLMs. In addition, EM-LLM outperforms its popular counterpart, RAG, in a wide range of tasks, while requiring similar resources. Notably, EM-LLM's performance even surpasses full-context models in most tasks, while successfully performing retrieval across 10 million tokens -- a scale computationally infeasible for such models. Finally, our analysis reveals strong correlations between EM-LLM's event segmentation and human-perceived events, suggesting parallels between this artificial system and its biological counterpart, thereby offering a novel computational framework for exploring human memory mechanisms. Zafeirios Fountas, Martin Benfeghoul, Adnan Oomerjee, Fenia Christopoulou, Gerasimos Lampouras, Haitham Bou-Ammar, Jun Wang 0012 |
ICLR | 7 |
| 2025 | Robust Function-Calling for On-Device Language Model via Function MaskingabstractLarge language models have demonstrated impressive value in performing as autonomous agents when equipped with external tools and API calls. Nonetheless, effectively harnessing their potential for executing complex tasks crucially relies on enhancements in their function-calling capabilities. This paper identifies a critical gap in existing function-calling models, where performance varies significantly across benchmarks, often due to over-fitting to specific naming conventions. To address such an issue, we introduce Hammer, a novel family of foundation models specifically engineered for on-device function calling. Hammer employs an augmented dataset that enhances models’ sensitivity to irrelevant functions and incorporates function masking techniques to minimize over-fitting. Our empirical evaluations reveal that Hammer not only outperforms larger models but also demonstrates robust generalization across diverse benchmarks, achieving state-of-the-art results. Our open-source contributions include a specialized dataset for irrelevance detection, a tuning framework for enhanced generalization, and the Hammer models, establishing a new standard for function-calling performance. Qiqiang Lin, Muning Wen, Qiuying Peng, Guanyu Nie, Junwei Liao, Xiaoyun Mo, Jiamu Zhou, Yin Zhao, Jun Wang 0012, Weinan Zhang 0001 |
ICLR | 11 |
| 2025 | On the Optimization Landscape of Low Rank Adaptation Methods for Large Language ModelsabstractTraining Large Language Models (LLMs) poses significant memory challenges, making low-rank adaptation methods an attractive solution. Previously, Low-Rank Adaptation (LoRA) addressed this by adding a trainable low-rank matrix to the frozen pre-trained weights in each layer, reducing the number of trainable parameters and optimizer states. GaLore, which compresses the gradient matrix instead of the weight matrix, has demonstrated superior performance to LoRA with faster convergence and reduced memory consumption. Despite their empirical success, the performance of these methods has not been fully understood or explained theoretically. In this paper, we analyze the optimization landscapes of LoRA, GaLore, and full-rank methods, revealing that GaLore benefits from fewer spurious local minima and a larger region that satisfies the \pl, a variant of Polyak-Łojasiewicz (PL) condition, leading to faster convergence. Our analysis leads to a novel method, GaRare, which further improves GaLore by using gradient random projection to reduce computational overhead. Practically, GaRare achieves strong performance in both pre-training and fine-tuning tasks, offering a more efficient approach to large-scale model adaptation. Xu-Hui Liu, Yali Du 0001, Jun Wang 0012, Yang Yu 0001 |
ICLR | 3 |
| 2025 | DistRL: An Asynchronous Distributed Reinforcement Learning Framework for On-Device Control AgentabstractOn-device control agents, especially on mobile devices, are responsible for operating mobile devices to fulfill users' requests, enabling seamless and intuitive interactions. Integrating Multimodal Large Language Models (MLLMs) into these agents enhances their ability to understand and execute complex commands, thereby improving user experience. However, fine-tuning MLLMs for on-device control presents significant challenges due to limited data availability and inefficient online training processes. This paper introduces DistRL, a novel framework designed to enhance the efficiency of online RL fine-tuning for mobile device control agents. DistRL employs centralized training and decentralized data acquisition to ensure efficient fine-tuning in the context of dynamic online interactions. Additionally, the framework is backed by our tailor-made RL algorithm, which effectively balances exploration with the prioritized utilization of collected data to ensure stable and robust training. Our experiments show that, on average, DistRL delivers a 3$\times$ improvement in training efficiency and enables training data collection 2.4$\times$ faster than the leading synchronous multi-machine methods. Notably, after training, DistRL achieves a 20\% relative improvement in success rate compared to state-of-the-art methods on general Android tasks from an open benchmark, significantly outperforming existing approaches while maintaining the same training time. These results validate DistRL as a scalable and efficient solution, offering substantial improvements in both training efficiency and agent performance for real-world, in-the-wild device control tasks. Taiyi Wang, Jianheng Liu, Jianye Hao, Jun Wang 0012, Kun Shao |
ICLR | 5 |
| 2025 | Mixture of Attentions For Speculative DecodingabstractThe growth in the number of parameters of Large Language Models (LLMs) has led to a significant surge in computational requirements, making them challenging and costly to deploy.
Speculative decoding (SD) leverages smaller models to efficiently propose future tokens, which are then verified by the LLM in parallel.
Small models that utilise activations from the LLM currently achieve the fastest decoding speeds.
However, we identify several limitations of SD models including the lack of on-policyness during training and partial observability.
To address these shortcomings, we propose a more grounded architecture for small models by introducing a Mixture of Attentions for SD.
Our novel architecture can be applied in two scenarios: a conventional single device deployment and a novel client-server deployment where the small model is hosted on a consumer device and the LLM on a server.
In a single-device scenario, we demonstrate state-of-the-art speedups improving EAGLE-2 by 9.5% and its acceptance length by 25%.
In a client-server setting, our experiments demonstrate: 1) state-of-the-art latencies with minimal calls to the server for different network conditions, and 2) in the event of a complete disconnection, our approach can maintain higher accuracy compared to other SD methods and demonstrates advantages over API calls to LLMs, which would otherwise be unable to continue the generation process. Matthieu Zimmer, Milan Gritta, Gerasimos Lampouras, Haitham Bou-Ammar, Jun Wang 0012 |
ICLR | 5 |
| 2025 | Multimodal Emotion Recognition in Conversations via Graph Structure LearningabstractMultimodal Emotion Recognition in Conversations (MERC) aims to detect emotions expressed in each utterance within conversational videos. Graph-based methods are widely employed in MERC due to their superiority in modeling intricate speaker-sensitive and context-sensitive dependencies in conversations. Despite promising advancements made, existing graph-based methods primarily suffer from two inherent issues due to their reliance on manually predefined graph structures: structural redundancy, which burdens models with irrelevant noise aggregation, and insufficient connections, which results in a lack of cross-modal contextual cues. To address the above issues, we propose a novel graph structure learning framework for MERC, which comprises two key components: Context-aware Graph Sparsification (CGS) and Implicit Graph Relation Mining (IGR). CGS employs an edge selection network to refine the manually predefined graph, filtering out noisy information caused by structural redundancy. IGR explores potential connections that are beneficial for emotional reasoning. Experimental results on two datasets show that our proposed framework significantly improves the performance of graph-based methods in MERC. Geng Tu, Yice Zhang, Jun Wang 0012, Bin Liang 0004, Yue Yu 0001, Min Yang 0007, Ruifeng Xu 0001 |
ICME | 4 |
| 2025 | ACCL: A Plug-and-play Adaptive Confusion-aware Contrastive Loss for UAV-to-Satellite GeolocalizationabstractUAV-to-Satellite geolocalization aims to estimate the location of an aerial-view query image taken by an Unmanned Aerial Vehicle (UAV) by matching it to satellite images annotated with known locations. However, it is difficult for existing methods to distinguish neighboring satellite images that exhibit a high degree of visual similarity. To address this issue, we introduce a plug-and-play adaptive confusion-aware contrastive loss (ACCL) to explicitly enhance the model’s discriminative ability, which gives more tolerance to high confusion query samples by means of elaborating a confusion metric function. As a plug-and-play loss module, ACCL can be easily incorporated into various UAV-to-Satellite geolocalization methods without additional modifications. To demonstrate the effectiveness of our proposed method, we conduct extensive experiments on one publicly available geolocalization dataset (i.e. NewYorkFly) and to further prove the effectiveness of our method in different scenarios, we collect two new geolocalization datasets (LasVegasFly and HollywoodFly), which contain drone-captured aerial images and dense sampled satellite images in various geomorphic regions. Experimental results indicate that our method can achieve an obvious performance improvement over the state-of-the-art methods on all three datasets. Our code and collected datasets are available at https://github.com/NWPU-CPS/ACCL. Yining Zhu, Jun Wang 0012, Boxuan Li, Long Xiao, Jikun Shen, Yuan Yao 0004 |
ICME | 3 |
| 2025 | Large Language Models are Demonstration Pre-Selectors for ThemselvesabstractIn-context learning with large language models (LLMs) delivers strong few-shot performance by choosing few-shot demonstrations from the entire training dataset. However, previous few-shot in-context learning methods, which calculate similarity scores for choosing demonstrations, incur high computational costs by repeatedly retrieving large-scale datasets for each query. This is due to their failure to recognize that not all demonstrations are equally informative, and many less informative demonstrations can be inferred from a core set of highly informative ones. To this end, we propose FEEDER (FEw yet Essential Demonstration prE-selectoR), a novel \emph{pre-selection} framework that identifies a core subset of demonstrations containing the most informative examples. This subset, referred to as the FEEDER set, consists of demonstrations that capture both the ''sufficiency'' and ''necessity'' information to infer the entire dataset. Notice that FEEDER is selected before the few-shot in-context learning, enabling more efficient few-shot demonstrations choosing in a smaller set. To identify FEEDER, we propose a novel effective tree based algorithm. Once selected, it can replace the original dataset, leading to improved efficiency and prediction accuracy in few-shot in-context learning. Additionally, FEEDER also benefit fine-tuning LLMs, we propose a bi-level optimization method enabling more efficient training without sacrificing performance when datasets become smaller. Our experiments are on 6 text classification datasets, 1 reasoning dataset, and 1 semantic-parsing dataset, across 6 LLMs (ranging from 335M to 7B parameters), demonstrate that: (i) In few-shot inference, FEEDER achieves superior (or comparable) performance while utilizing only half the input training data. (ii) In fine-tuning, FEEDER significantly boosts the performance of LLMs. Jiarui Jin, Haoxuan Li 0001, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012, Mengyue Yang |
ICML | 8 |
| 2025 | Constrain Alignment with Sparse AutoencodersabstractThe alignment of large language models (LLMs) with human preferences remains a key challenge. While post-training techniques like Reinforcement Learning from Human Feedback (RLHF) and Direct Preference Optimization (DPO) have achieved notable success, they often experience computational inefficiencies and training instability. In this paper, we propose Feature-level constrained Preference Optimization (FPO), a novel method designed to simplify the alignment process while ensuring stability. FPO leverages pre-trained Sparse Autoencoders (SAEs) and introduces feature-level constraints, allowing for efficient, sparsity-enforced alignment. Our approach enjoys efficiency by using sparse features activated in a well-trained sparse autoencoder and the quality of sequential KL divergence by using the feature-level offline reference. Experimental results on benchmark datasets demonstrate that FPO achieves an above 5% absolute improvement in win rate with much lower computational cost compared to state-of-the-art baselines, making it a promising solution for efficient and controllable LLM alignments. Qingyu Yin, Chak Tou Leong, Minjun Zhu, Hanqi Yan, Qiang Zhang 0026, Yulan He 0001, Wenjie Li 0002, Jun Wang 0012, Yue Zhang 0004, Linyi Yang |
ICML | 9 |
| 2025 | Taming Multi-Agent Reinforcement Learning with Estimator Variance Reduction
Taher Jafferjee, Juliusz Krysztof Ziomek, Tianpei Yang, Zipeng Dai, Matthew E. Taylor, Kun Shao, Jun Wang 0012, David Mguni |
AAMAS | 8 |
| 2025 | Unlocking the Potential of Decentralized LLM-based MAS: Privacy Preservation and Monetization in Collective Intelligence
Yingxuan Yang, Qiuying Peng, Jun Wang 0012, Ying Wen 0001, Weinan Zhang 0001 |
AAMAS | 3 |
| 2025 | Mean Field Correlated Imitation Learning
Chengdong Ma, Qirui Mi, Ning Yang 0005, Mengyue Yang, Haifeng Zhang 0002, Jun Wang 0012, Yaodong Yang 0001 |
AAMAS | 8 |
| 2025 | Optimizing Case-Based Reasoning System for Functional Test Script Generation with Large Language ModelsabstractIn this work, we explore the potential of large language models (LLMs) for generating functional test scripts, which necessitates understanding the dynamically evolving code structure of the target software. To achieve this, we propose a case-based reasoning (CBR) system utilizing a 4R cycle (i.e., retrieve, reuse, revise, and retain), which maintains and leverages a case bank of test intent descriptions and corresponding test scripts to facilitate LLMs for test script generation. To improve user experience further, we introduce Re4, an optimization method for the CBR system, comprising reranking-based retrieval finetuning and reinforced reuse finetuning. Specifically, we first identify positive examples with high semantic and script similarity, providing reliable pseudo-labels for finetuning the retriever model without costly labeling. Then, we apply supervised finetuning, followed by a reinforcement learning finetuning stage, to align LLMs with our production scenarios, ensuring the faithful reuse of retrieved cases. Extensive experimental results on two product development units from Huawei Datacom demonstrate the superiority of the proposed CBR+Re4. Notably, we also show that the proposed Re4 method can help alleviate the repetitive generation issues with LLMs. Siyuan Guo 0001, Huiwu Liu, Yuming Xie, Hechang Chen, Yi Chang 0001, Jun Wang 0012 |
KDD (2) | 9 |
| 2025 | Why Not Together? A Multiple-Round Recommender System for Queries and Items
Jiarui Jin, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
KDD (1) | 5 |
| 2025 | Uncertainty-quantified Rollout Policy Adaptation for Unlabelled Cross-domain Video Temporal GroundingabstractVideo Temporal Grounding (TG) aims to temporally locate video segments matching a natural language description (a query) in a long video. While Vision-Language Models (VLMs) are effective at holistic semantic matching, they often struggle with fine-grained temporal
localisation. Recently, Group Relative Policy Optimisation (GRPO) reformulates the inference process as a reinforcement learning task, enabling fine-grained grounding and achieving strong in-domain performance. However, GRPO relies on labelled data, making it unsuitable in unlabelled domains. Moreover, because videos are large and expensive to store and process, performing full-scale adaptation introduces prohibitive latency and computational overhead, making it impractical for real-time deployment. To overcome both problems, we introduce a Data-Efficient Unlabelled Cross-domain Temporal Grounding method, from which a model is first trained on a labelled source domain, then adapted to a target domain using only a small number of {\em unlabelled videos from the target domain}. This approach eliminates the need for target annotation and keeps both computational and storage overhead low enough to run in real time. Specifically, we introduce \textbf{U}ncertainty-quantified \textbf{R}ollout \textbf{P}olicy \textbf{A}daptation (\textbf{URPA}) for cross-domain knowledge transfer in learning video temporal grounding without target labels. URPA generates multiple candidate predictions using GRPO rollouts, averages them to form a pseudo label, and estimates confidence from the variance across these rollouts. This confidence then weights the training rewards, guiding the model to focus on reliable supervision. Experiments on three datasets across six cross-domain settings show that URPA generalises well using only a few unlabelled target videos. Codes are given in supplemental materials. Zixu Cheng, Shaogang Gong, Isabel Guan, Jianye Hao, Jun Wang 0012, Kun Shao |
NeurIPS | 6 |
| 2025 | ThinkBench: Dynamic Out-of-Distribution Evaluation for Robust LLM ReasoningabstractEvaluating large language models (LLMs) poses significant challenges, particularly due to issues of data contamination and the leakage of correct answers. To address these challenges, we introduce ThinkBench, a novel evaluation framework designed to robustly evaluate the reasoning capability of LLMs. ThinkBench proposes a dynamic data generation method for constructing out-of-distribution (OOD) datasets and offers an OOD dataset that contains 2,912 samples drawn from reasoning tasks. ThinkBench unifies the evaluation of reasoning models and non-reasoning models. We evaluate 16 LLMs and 4 PRMs under identical experimental conditions and show that most of the LLMs' performance are far from robust and they face a certain level of data leakage. By dynamically generating OOD datasets, ThinkBench effectively provides a reliable evaluation of LLMs and reduces data contamination impact. Our data and codes are available at https://github.com/huangshulin123/ThinkBench. Shulin Huang, Linyi Yang, Yan Song 0003, Shawn Chen, Leyang Cui, Ziyu Wan, Qingcheng Zeng, Ying Wen 0001, Kun Shao, Weinan Zhang 0001, Jun Wang 0012, Yue Zhang 0004 |
NeurIPS | 11 |
| 2025 | MobileUse: A Hierarchical Reflection-Driven GUI Agent for Autonomous Mobile OperationabstractRecent advances in Multimodal Large Language Models (MLLMs) have enabled the development of mobile agents that can understand visual inputs and follow user instructions, unlocking new possibilities for automating complex tasks on mobile devices. However, applying these models to real-world mobile scenarios remains a significant challenge due to the long-horizon task execution, difficulty in error recovery, and the cold-start problem in unfamiliar environments. To address these challenges, we propose MobileUse, a GUI agent designed for robust and adaptive mobile task execution. To improve resilience in long-horizon tasks and dynamic environments, we introduce a hierarchical reflection architecture that enables the agent to self-monitor, detect, and recover from errors across multiple temporal scales—ranging from individual actions to overall task completion—while maintaining efficiency through a Reflection-on-Demand strategy. To tackle cold-start issues, we further introduce a proactive exploration module, which enriches the agent’s understanding of the environment through self-planned exploration. Evaluations on the AndroidWorld and AndroidLab benchmarks demonstrate that MobileUse establishes new state-of-the-art performance, achieving success rates of 62.9% and 44.2%, respectively. To facilitate real-world applications, we release an out-of-the-box toolkit for automated task execution on physical mobile devices, which is available at https://github.com/MadeAgents/mobile-use. Ning Li 0029, Xiangmou Qu, Jiamu Zhou, Muning Wen, Kounianhua Du, Xingyu Lou, Qiuying Peng, Jun Wang 0012, Weinan Zhang 0001 |
NeurIPS | 8 |
| 2025 | A Principle of Targeted Intervention for Multi-Agent Reinforcement LearningabstractSteering cooperative multi-agent reinforcement learning (MARL) towards desired outcomes is challenging, particularly when the global guidance from a human on the whole multi-agent system is impractical in a large-scale MARL. On the other hand, designing external mechanisms (e.g., intrinsic rewards and human feedback) to coordinate agents mostly relies on empirical studies, lacking a easy-to-use research tool. In this work, we employ multi-agent influence diagrams (MAIDs) as a graphical framework to address the above issues. First, we introduce the concept of MARL interaction paradigms (orthogonal to MARL learning paradigms), using MAIDs to analyze and visualize both unguided self-organization and global guidance mechanisms in MARL. Then, we design a new MARL interaction paradigm, referred to as the targeted intervention paradigm that is applied to only a single targeted agent, so the problem of global guidance can be mitigated. In implementation, we introduce a causal inference technique—referred to as Pre-Strategy Intervention (PSI)—to realize the targeted intervention paradigm. Since MAIDs can be regarded as a special class of causal diagrams, a composite desired outcome that integrates the primary task goal and an additional desired outcome can be achieved by maximizing the corresponding causal effect through the PSI. Moreover, the bundled relevance graph analysis of MAIDs provides a tool to identify whether an MARL learning paradigm is workable under the design of an MARL interaction paradigm. In experiments, we demonstrate the effectiveness of our proposed targeted intervention, and verify the result of relevance graph analysis. Anjie Liu, Samuel Kaski, Jun Wang 0012, Mengyue Yang |
NeurIPS | 4 |
| 2025 | EconGym: A Scalable AI Testbed with Diverse Economic TasksabstractArtificial intelligence (AI) has become a powerful tool for economic research, enabling large-scale simulation and policy optimization. However, applying AI effectively requires simulation platforms for scalable training and evaluation—yet existing environments remain limited to simplified, narrowly scoped tasks, falling short of capturing complex economic challenges such as demographic shifts, multi-government coordination, and large-scale agent interactions.To address this gap, we introduce EconGym, a scalable and modular testbed that connects diverse economic tasks with AI algorithms. Grounded in rigorous economic modeling, EconGym implements 11 heterogeneous role types (e.g., households, firms, banks, governments), their interaction mechanisms, and agent models with well-defined observations, actions, and rewards. Users can flexibly compose economic roles with diverse agent algorithms to simulate rich multi-agent trajectories across 25+ economic tasks for AI-driven policy learning and analysis.Experiments show that EconGym supports diverse and cross-domain tasks—such as coordinating fiscal, pension, and monetary policies—and enables benchmarking across AI, economic methods, and hybrids. Results indicate that richer task composition and algorithm diversity expand the policy space, while AI agents guided by classical economic methods perform best in complex settings. EconGym also scales to 100k agents with high realism and efficiency. Qirui Mi, Qipeng Yang, Zijun Fan, Wentian Fan, Heyang Ma, Chengdong Ma, Si-Yu Xia, Bo An 0001, Jun Wang 0012, Haifeng Zhang 0002 |
NeurIPS | 9 |
| 2025 | MF-LLM: Simulating Population Decision Dynamics via a Mean-Field Large Language Model FrameworkabstractSimulating collective decision-making involves more than aggregating individual behaviors; it emerges from dynamic interactions among individuals. While large language models (LLMs) offer strong potential for social simulation, achieving quantitative alignment with real-world data remains a key challenge. To bridge this gap, we propose the \textbf{M}ean-\textbf{F}ield \textbf{LLM} (\textbf{MF-LLM}) framework, the first to incorporate mean field theory into LLM-based social simulation. MF-LLM models bidirectional interactions between individuals and the population through an iterative process, generating population signals to guide individual decisions, which in turn update the signals. This interplay produces coherent trajectories of collective behavior.
To improve alignment with real-world data, we introduce \textbf{IB-Tune}, a novel fine-tuning method inspired by the \textbf{I}nformation \textbf{B}ottleneck principle, which retains population signals most predictive of future actions while filtering redundant history. Evaluated on a real-world social dataset, MF-LLM reduces KL divergence to human population distributions by \textbf{47\%} compared to non-mean-field baselines, enabling accurate trend forecasting and effective intervention planning.
Generalizing across 7 domains and 4 LLM backbones, MF-LLM provides a scalable, high-fidelity foundation for social simulation. Qirui Mi, Mengyue Yang, Xiangning Yu 0001, Cheng Deng 0001, Bo An 0001, Haifeng Zhang 0002, Xu Chen 0017, Jun Wang 0012 |
NeurIPS | 9 |
| 2025 | Succeed or Learn Slowly: Sample Efficient Off-Policy Reinforcement Learning for Mobile App ControlabstractReinforcement learning (RL) using foundation models for policy approximations in multi-turn tasks remains challenging. We identify two main limitations related to sparse reward settings and policy gradient updates, based on which we formulate a key insight: updates from positive samples with high returns typically do not require policy regularisation, whereas updates from negative samples, reflecting undesirable behaviour, can harm model performance. This paper introduces Succeed or Learn Slowly (SoLS), a novel off-policy RL algorithm evaluated on mobile app control tasks. SoLS improves sample efficiency when fine-tuning foundation models for user interface navigation via a modified off-policy actor-critic approach, applying direct policy updates for positive samples and conservative, regularised updates for negative ones to prevent model degradation. We augment SoLS with Successful Transition Replay (STR), which prioritises learning from successful interactions, further improving sample efficiency. We evaluate SoLS on the AndroidWorld benchmark, where it significantly outperforms existing methods (at least 17\% relative increase), including prompt-engineering and RL approaches, while requiring substantially fewer computational resources than GPT-4o-based methods with 5-60x faster inference. Georgios Papoudakis, Thomas Coste, Jianye Hao, Jun Wang 0012, Kun Shao |
NeurIPS | 4 |
| 2025 | ReMA: Learning to Meta-Think for LLMs with Multi-agent Reinforcement LearningabstractRecent research on Reasoning of Large Language Models (LLMs) has sought to further enhance their performance by integrating meta-thinking—enabling models to monitor, evaluate, and control their reasoning processes for more adaptive and effective problem-solving.
However, current single-agent work lacks a specialized design for acquiring meta-thinking, resulting in low efficacy.
To address this challenge, we introduce Reinforced Meta-thinking Agents (ReMA), a novel framework that leverages Multi-Agent Reinforcement Learning (MARL) to elicit meta-thinking behaviors, encouraging LLMs to think about thinking.
ReMA decouples the reasoning process into two hierarchical agents: a high-level meta-thinking agent responsible for generating strategic oversight and plans, and a low-level reasoning agent for detailed executions.
Through iterative reinforcement learning with aligned objectives, these agents explore and learn collaboration, leading to improved generalization and robustness.
Empirical results from single-turn experiments demonstrate that ReMA outperforms single-agent RL baselines on complex reasoning tasks, including competitive-level mathematical benchmarks and LLM-as-a-Judge benchmarks.
Additionally, we further extend ReMA to multi-turn interaction settings, leveraging turn-level ratio and parameter sharing to improve efficiency.
Comprehensive ablation studies further illustrate the evolving dynamics of each distinct agent, providing valuable insights into how the meta-thinking reasoning process enhances the reasoning capabilities of LLMs. Ziyu Wan, Xiaoyu Wen 0001, Yan Song 0003, Hanjing Wang, Linyi Yang, Mark Schmidt 0001, Jun Wang 0012, Weinan Zhang 0001, Shuyue Hu, Ying Wen 0001 |
NeurIPS | 8 |
| 2025 | CMoB: Modality Valuation via Causal Effect for Balanced Multimodal LearningabstractExisting early and late fusion frameworks in multimodal learning are confronted with the fundamental challenge of modality imbalance, wherein disparities in representational capacities induce inter-modal competition during training. Current research methodologies primarily rely on modality-level contribution assessments to measure gaps in representational capabilities and enhance poorly learned modalities, overlooking the dynamic variations of modality contributions across individual samples. To address this, we propose a Causal-aware Modality valuation approach for Balanced multimodal learning (CMoB). We define a benefit function based on Shannon's theory of informational uncertainty to evaluate the changes in the importance of samples across different stages of multimodal training. Inspired by human cognitive science, we propose a causal-aware modality contribution quantification method from a causal perspective to capture fine-grained changes in modality contribution degrees within samples. In the iterative training of multimodal learning, we develop targeted modal enhancement strategies that dynamically select and optimize modalities based on real-time evaluation of their contribution variations across training samples. Our method enhances the discriminative ability of key modalities and the learning capacity of weak modalities while achieving fine-grained balance in multimodal learning. Extensive experiments on benchmark multimodal datasets and multimodal frameworks demonstrate the superiority of our CMoB approach for balanced multimodal learning. Jun Wang 0012, Fuyuan Cao, Zhixin Xue, Xingwang Zhao 0001, Jiye Liang |
NeurIPS | 1 |
| 2025 | Causal Sufficiency and Necessity Improves Chain-of-Thought ReasoningabstractChain-of-Thought (CoT) prompting plays an indispensable role in endowing large language models (LLMs) with complex reasoning capabilities. However, CoT currently faces two fundamental challenges: (1) Sufficiency, which ensures that the generated intermediate inference steps comprehensively cover and substantiate the final conclusion; and (2) Necessity, which identifies the inference steps that are truly indispensable for the soundness of the resulting answer. We propose a causal framework that characterizes CoT reasoning through the dual lenses of sufficiency and necessity. Incorporating causal Probability of Sufficiency and Necessity allows us not only to determine which steps are logically sufficient or necessary to the prediction outcome, but also to quantify their actual influence on the final reasoning outcome under different intervention scenarios, thereby enabling the automated addition of missing steps and the pruning of redundant ones. Extensive experimental results on various mathematical and commonsense reasoning benchmarks confirm substantial improvements in reasoning efficiency and reduced token usage without sacrificing accuracy. Our work provides a promising direction for improving LLM reasoning performance and cost-effectiveness. The code will be publicly available upon acceptance at: https://anonymous.4open.science/r/causalmath-1CEF. Xiangning Yu 0001, Zhuohan Wang, Linyi Yang, Haoxuan Li 0001, Anjie Liu, Xiao Xue 0001, Jun Wang 0012, Mengyue Yang |
NeurIPS | 7 |
| 2025 | Self-Verifying Reflection Helps Transformers with CoT ReasoningabstractAdvanced large language models (LLMs) frequently reflect in reasoning chain-of-thoughts (CoTs), where they self-verify the correctness of current solutions and explore alternatives. However, given recent findings that LLMs detect limited errors in CoTs, how reflection contributes to empirical improvements remains unclear. To analyze this issue, in this paper, we present a minimalistic reasoning framework to support basic self-verifying reflection for small transformers without natural language, which ensures analytic clarity and reduces the cost of comprehensive experiments. Theoretically, we prove that self-verifying reflection guarantees improvements if verification errors are properly bounded. Experimentally, we show that tiny transformers, with only a few million parameters, benefit from self-verification in both training and reflective execution, reaching remarkable LLM-level performance in integer multiplication and Sudoku. Similar to LLM results, we find that reinforcement learning (RL) improves in-distribution performance and incentivizes frequent reflection for tiny transformers, yet RL mainly optimizes shallow statistical patterns without faithfully reducing verification errors. In conclusion, integrating generative transformers with discriminative verification inherently facilitates CoT reasoning, regardless of scaling and natural language. Zhongwei Yu, Wannian Xia, Bo Xu 0002, Haifeng Zhang 0002, Yali Du 0001, Jun Wang 0012 |
NeurIPS | 7 |
| 2025 | Risk-aware Direct Preference Optimization under Nested Risk MeasureabstractWhen fine-tuning pre-trained Large Language Models (LLMs) to align with human values and intentions, maximizing the estimated reward can lead to superior performance, but it also introduces potential risks due to deviations from the reference model's intended behavior. Most existing methods typically introduce KL divergence to constrain deviations between the trained model and the reference model; however, this may not be sufficient in certain applications that require tight risk control. In this paper, we introduce Risk-aware Direct Preference Optimization (Ra-DPO), a novel approach that incorporates risk-awareness by employing a class of nested risk measures. This approach formulates a constrained risk-aware advantage function maximization problem and then converts the Bradley-Terry model into a token-level representation. The objective function maximizes the likelihood of the policy while suppressing the deviation between a trained model and the reference model using a sequential risk ratio, thereby enhancing the model's risk-awareness. Experimental results across three open-source datasets: IMDb Dataset, Anthropic HH Dataset, and AlpacaEval, demonstrate the proposed method's superior performance in balancing alignment performance and model drift. Lin Li 0090, Yajie Qi, Huizhong Song, Yaodong Yang 0001, Jun Wang 0012, Wei Wei 0018 |
NeurIPS | 6 |
| 2025 | Curious Causality-Seeking Agents in Open-ended WorldsabstractWhen building a world model, a common assumption is that the environment has a single, unchanging underlying causal rule, like applying Newton's laws to every situation. However, in truly open-ended environments, the apparent causal mechanism may drift over time because the agent continually encounters novel contexts and operates within a limited observational window. This brings about a problem that, when building a world model, even subtle shifts in policy or environment states can alter the very observed causal mechanisms.
In this work, we introduce the Meta-Causal Graph as world models for open-ended environments, a minimal unified representation that efficiently encodes the transformation rules governing how causal structures shift across different latent world states. A single Meta-Causal Graph is composed of multiple causal subgraphs, each triggered by meta state, which is in the latent state space. Building on this representation, we introduce a Causality-Seeking Agent whose objectives are to (1) identify the meta states that trigger each subgraph, (2) discover the corresponding causal relationships by agent curiosity-driven intervention policy, and (3) iteratively refine the Meta-Causal Graph through ongoing curiosity-driven exploration and agent experiences. Experiments on both synthetic tasks and a challenging robot arm manipulation task demonstrate that our method robustly captures shifts in causal dynamics and generalizes effectively to previously unseen contexts. Haoxuan Li 0001, Haifeng Zhang 0002, Jun Wang 0012, Francesco Faccio, Jürgen Schmidhuber, Mengyue Yang |
NeurIPS | 4 |
| 2025 | An enhanced 6G pervasive channel model towards standardization
Cheng-Xiang Wang 0001, Zhen Lv 0002, Chen Huang 0004, Yusong Huang, Jun Wang 0012, Jie Huang 0004, Xiaohu You 0001 |
Sci. China Inf. Sci. | 5 |
| 2025 | GARNN: An interpretable graph attentive recurrent neural network for predicting blood glucose levels via multivariate time seriesabstractAccurate prediction of future blood glucose (BG) levels can effectively improve BG management for people living with type 1 or 2 diabetes, thereby reducing complications and improving quality of life. The state of the art of BG prediction has been achieved by leveraging advanced deep learning methods to model multimodal data, i.e., sensor data and self-reported event data, organized as multi-variate time series (MTS). However, these methods are mostly regarded as "black boxes" and not entirely trusted by clinicians and patients. In this paper, we propose interpretable graph attentive recurrent neural networks (GARNNs) to model MTS, explaining variable contributions via summarizing variable importance and generating feature maps by graph attention mechanisms instead of post-hoc analysis. We evaluate GARNNs on four datasets, representing diverse clinical scenarios. Upon comparison with fifteen well-established baseline methods, GARNNs not only achieve the best prediction accuracy but also provide high-quality temporal interpretability, in particular for postprandial glucose levels as a result of corresponding meal intake and insulin injection. These findings underline the potential of GARNN as a robust tool for improving diabetes care, bridging the gap between deep learning technology and real-world healthcare solutions. Chengzhe Piao, Taiyu Zhu, Stephanie E. Baldeweg, Pantelis Georgiou, Jun Wang 0012, Kezhi Li |
Neural Networks | 7 |
| 2025 | Robust Multiobjective Reinforcement Learning Considering Environmental UncertaintiesabstractNumerous real-world decision or control problems involve multiple conflicting objectives whose relative importance (preference) is required to be weighed in different scenarios. While Pareto optimality is desired, environmental uncertainties (e.g., environmental changes or observational noises) may mislead the agent into performing suboptimal policies. In this article, we present a novel multiobjective optimization paradigm, robust multiobjective reinforcement learning (RMORL) considering environmental uncertainties, to train a single model that can approximate robust Pareto-optimal policies across the entire preference space. To enhance policy robustness against environmental changes, an environmental disturbance is modeled as an adversarial agent across the entire preference space via incorporating a zero-sum game into a multiobjective Markov decision process (MOMDP). Additionally, we devise an adversarial defense technique against observational perturbations, which ensures that policy variations, perturbed by adversarial attacks on state observations, remain within bounds under any specified preferences. The proposed technique is assessed in five multiobjective environments with continuous action spaces, showcasing its effectiveness through comparisons with competitive baselines, which encompass classical and state-of-the-art schemes. Xiangkun He, Jianye Hao, Xu Chen 0017, Jun Wang 0012, Xuewu Ji, Chen Lv 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 4 |
| 2025 | User Behavior Simulation with Large Language Model-based AgentsabstractSimulating high quality user behavior data has always been a fundamental yet challenging problem in human-centered applications such as recommendation systems, social networks, among many others. The major difficulty of user behavior simulation originates from the intricate mechanism of human cognitive and decision processes. Recently, substantial evidence has suggested that by learning huge amounts of web knowledge, large language models (LLMs) can achieve human-like intelligence and generalization capabilities. Inspired by such capabilities, in this article, we take an initial step to study the potential of using LLMs for user behavior simulation in the recommendation domain. To make LLMs act like humans, we design profile, memory and action modules to equip them, building LLM-based agents to simulate real users. To enable interactions between different agents and observe their behavior patterns, we design a sandbox environment, where each agent can interact with the recommendation system, and different agents can converse with their friends via one-to-one chatting or one-to-many social broadcasting. In the experiments, we first demonstrate the believability of the agent-generated behaviors based on both subjective and objective evaluations. Then, to show the potential applications of our method, we simulate and study two social phenomena including (1) information cocoons and (2) user conformity behaviors. We find that controlling the personalization degree of recommendation algorithms and improving the heterogeneity of user social relations can be two effective strategies for alleviating the problem of information cocoon, and the conformity behaviors can be highly influenced by the amount of user social relations. To advance this direction, we have released our project at https://github.com/RUC-GSAI/YuLan-Rec . Lei Wang 0198, Jingsen Zhang, Hao Yang 0045, Jiakai Tang, Zeyu Zhang 0007, Xu Chen 0017, Yankai Lin 0001, Hao Sun 0002, Ruihua Song, Wayne Xin Zhao, Jun Xu 0001, Zhicheng Dou, Jun Wang 0012, Ji-Rong Wen |
ACM Trans. Inf. Syst. | 14 |
| 2024 | Large Language Models Are Neurosymbolic ReasonersabstractA wide range of real-world applications is characterized by their symbolic nature, necessitating a strong capability for symbolic reasoning. This paper investigates the potential application of Large Language Models (LLMs) as symbolic reasoners. We focus on text-based games, significant benchmarks for agents with natural language capabilities, particularly in symbolic tasks like math, map reading, sorting, and applying common sense in text-based worlds. To facilitate these agents, we propose an LLM agent designed to tackle symbolic challenges and achieve in-game objectives. We begin by initializing the LLM agent and informing it of its role. The agent then receives observations and a set of valid actions from the text-based games, along with a specific symbolic module. With these inputs, the LLM agent chooses an action and interacts with the game environments. Our experimental results demonstrate that our method significantly enhances the capability of LLMs as automated agents for symbolic reasoning, and our LLM agent is effective in text-based games involving symbolic tasks, achieving an average performance of 88% across all tasks. Shilong Deng, Yudi Zhang 0006, Zijing Shi, Ling Chen 0006, Mykola Pechenizkiy, Jun Wang 0012 |
AAAI | 7 |
| 2024 | Human-Guided Moral Decision Making in Text-Based GamesabstractTraining reinforcement learning (RL) agents to achieve desired goals while also acting morally is a challenging problem. Transformer-based language models (LMs) have shown some promise in moral awareness, but their use in different contexts is problematic because of the complexity and implicitness of human morality. In this paper, we build on text-based games, which are challenging environments for current RL agents, and propose the HuMAL (Human-guided Morality Awareness Learning) algorithm, which adaptively learns personal values through human-agent collaboration with minimal manual feedback. We evaluate HuMAL on the Jiminy Cricket benchmark, a set of text-based games with various scenes and dense morality annotations, using both simulated and actual human feedback. The experimental results demonstrate that with a small amount of human feedback, HuMAL can improve task performance and reduce immoral behavior in a variety of games and is adaptable to different personal values. Zijing Shi, Ling Chen 0006, Yali Du 0001, Jun Wang 0012 |
AAAI | 5 |
| 2024 | Looking Ahead to Avoid Being Late: Solving Hard-Constrained Traveling Salesman ProblemabstractMany real-world problems can be formulated as a constrained Traveling Salesman Problem (TSP).However, the constraints are always complex and numerous, making the TSPs challenging to solve.When the number of complicated constraints grows, it is timeconsuming for traditional heuristic algorithms to avoid illegitimate outcomes.Learning-based methods provide an alternative to solve TSPs in a soft manner, which also supports GPU acceleration to generate solutions quickly.Nevertheless, the soft manner inevitably results in difficulty solving hard-constrained problems with learning algorithms, and the conflicts between legality and optimality may substantially affect the optimality of the solution.To overcome this problem and to have an effective solution against hard constraints, we proposed a novel learning-based method, MUSLA, that uses multi-step looking-ahead information as the feature to improve the legality of TSP with Time Windows (TSPTW) solutions.Besides, we constructed TSPTW datasets with hard constraints in order to accurately evaluate and benchmark the statistical performance of various approaches, which can serve the community for future research.With comprehensive experiments on diverse datasets, MUSLA outperforms existing baselines and shows generalizability potential. Jingxiao Chen, Ziqin Gong, Lvda Chen, Minghuan Liu, Jun Wang 0012, Yong Yu 0001, Weinan Zhang 0001 |
DAI | 5 |
| 2024 | Adaptive Command : Real-Time Policy Adjustment via Language Models in StarCraft IIabstractWe present Adaptive Command, a novel framework integrating large language models (LLMs) with behavior trees for real-time strategic decision-making in StarCraft II.Our system focuses on enhancing human-AI collaboration in complex, dynamic environments through natural language interactions.The framework comprises: (1) an LLM-based strategic advisor, (2) a behavior tree for action execution, and (3) a natural language interface with speech capabilities.User studies demonstrate significant improvements in player decision-making and strategic adaptability, particularly benefiting novice players and those with disabilities.This work contributes to the field of real-time human-AI collaborative decisionmaking, offering insights applicable beyond RTS games to various complex decision-making scenarios. Weiyu Ma, Shu Lin 0003, Haifeng Zhang 0002, Jun Wang 0012 |
DAI | 5 |
| 2024 | Incremental Model Quantization for Federated Learning in Wireless NetworksabstractFederated Learning (FL) has been widely recognized as a promising promoter for future intelligent wireless networks, by collaboratively training a global machine learning (ML) model in a privacy-preserving manner. However, the transmission of large-scale models between clients and servers is susceptible to the limitation of available communication resources which deteriorates FL performance. Some recently proposed model quantization techniques can effectively reduce communication costs by compressing the amount of model data to be transmitted. Unfortunately, conventional quantization methods face difficulties in wireless with rapidly changing channels. In this paper, we propose a federated learning scheme with incremental model quantization and uploading mechanism, called Fed_IQ. Specifically, individual clients quantize the gradients of the local model to obtain base gradients as well as incremental gradients and send them to the server. Then the server combines the base and incremental gradients to obtain a more accurate global model where the quantization levels of gradients can be adaptively adjusted according to the instantaneous channel states. Experimental results show our proposed Fed_IQ can significantly reduce transmission delay and improve model accuracy in a wireless network compared with state-of-the-art algorithms. Gang Feng 0004, Yijing Liu 0001, Jun Wang 0012 |
GLOBECOM | 6 |
| 2024 | RACon: Retrieval-Augmented Simulated Character Locomotion ControlabstractIn computer animation, driving a simulated character with lifelike motion is challenging. Current generative models, though able to generalize to diverse motions, often pose challenges to the responsiveness of end-user control. To address these issues, we introduce RACon: Retrieval-Augmented Simulated Character Locomotion Control. Our end-to-end hierarchical reinforcement learning method utilizes a retriever and a motion controller. The retriever searches motion experts from a user-specified database in a task-oriented fashion, which boosts the responsiveness to the user’s control. The selected motion experts and the manipulation signal are then transferred to the controller to drive the simulated character. In addition, a retrieval-augmented discriminator is designed to stabilize the training process. Our method surpasses existing techniques in both quality and quantity in locomotion control, as demonstrated in our empirical study. Moreover, by switching extensive databases for retrieval, it can adapt to distinctive motion types at run time. We will release our code upon acceptance. Yuxuan Mu, Shihao Zou, Kangning Yin, Zheng Tian 0002, Li Cheng 0001, Weinan Zhang 0001, Jun Wang 0012 |
ICME | 7 |
| 2024 | DS-Agent: Automated Data Science by Empowering Large Language Models with Case-Based ReasoningabstractIn this work, we investigate the potential of large language models (LLMs) based agents to automate data science tasks, with the goal of comprehending task requirements, then building and training the best-fit machine learning models. Despite their widespread success, existing LLM agents are hindered by generating unreasonable experiment plans within this scenario. To this end, we present DS-Agent, a novel automatic framework that harnesses LLM agent and case-based reasoning (CBR). In the development stage, DS-Agent follows the CBR framework to structure an automatic iteration pipeline, which can flexibly capitalize on the expert knowledge from Kaggle, and facilitate consistent performance improvement through the feedback mechanism. Moreover, DS-Agent implements a low-resource deployment stage with a simplified CBR paradigm to adapt past successful solutions from the development stage for direct code generation, significantly reducing the demand on foundational capabilities of LLMs. Empirically, DS-Agent with GPT-4 achieves 100% success rate in the development stage, while attaining 36% improvement on average one pass rate across alternative LLMs in the deployment stage. In both stages, DS-Agent achieves the best rank in performance, costing $1.60 and \$0.13 per run with GPT-4, respectively. Our data and code are open-sourced at https://github.com/guosyjlu/DS-Agent. Siyuan Guo 0001, Cheng Deng 0001, Ying Wen 0001, Hechang Chen, Yi Chang 0001, Jun Wang 0012 |
ICML | 6 |
| 2024 | AlphaZero-Like Tree-Search can Guide Large Language Model Decoding and TrainingabstractRecent works like Tree-of-Thought (ToT) and Reasoning via Planning (RAP) aim to augment the multi-step reasoning capabilities of LLMs by using tree-search algorithms. These methods rely on prompting a pre-trained model to serve as a value function and focus on problems with low search depth. As a result, these methods cannot benefit from in-domain training and only rely on pretraining process — they will not work in domains where the pre-trained LLM does not have enough knowledge to serve as an effective value function or in domains that require long-horizon planning. To address these limitations, we present an AlphaZero-like tree-search learning framework for LLMs (termed TS-LLM), systematically illustrating how tree-search with a learned value function can guide LLM decoding. TS-LLM distinguishes itself in two key ways. (1) Leveraging a learned value function and AlphaZero-like algorithms, our approach can be generally adaptable to a wide range of tasks, language models of any size, and tasks of varying search depths. (2) Our approach can guide LLMs during both inference and training, iteratively improving the LLMs. Empirical results across reasoning, planning, alignment, and decision-making tasks show that TS-LLM outperforms existing approaches and can handle trees with a depth of 64. Ziyu Wan, Xidong Feng, Muning Wen, Stephen McAleer, Ying Wen 0001, Weinan Zhang 0001, Jun Wang 0012 |
ICML | 7 |
| 2024 | Token-level Direct Preference OptimizationabstractFine-tuning pre-trained Large Language Models (LLMs) is essential to align them with human values and intentions. This process often utilizes methods like pairwise comparisons and KL divergence against a reference LLM, focusing on the evaluation of full answers generated by the models. However, the generation of these responses occurs in a token level, following a sequential, auto-regressive fashion. In this paper, we introduce Token-level Direct Preference Optimization (TDPO), a novel approach to align LLMs with human preferences by optimizing policy at the token level. Unlike previous methods, which face challenges in divergence efficiency, TDPO integrates forward KL divergence constraints for each token, improving alignment and diversity. Utilizing the Bradley-Terry model for a token-based reward system, our method enhances the regulation of KL divergence, while preserving simplicity without the need for explicit reward modeling. Experimental results across various text tasks demonstrate TDPO’s superior performance in balancing alignment with generation diversity. Notably, fine-tuning with TDPO strikes a better balance than DPO in the controlled sentiment generation and single-turn dialogue datasets, and significantly improves the quality of generated responses compared to both DPO and PPO-based RLHF methods. Yongcheng Zeng, Weiyu Ma, Ning Yang 0005, Haifeng Zhang 0002, Jun Wang 0012 |
ICML | 6 |
| 2024 | Language and Sketching: An LLM-driven Interactive Multimodal Multitask Robot Navigation FrameworkabstractThe socially-aware navigation system has evolved to adeptly avoid various obstacles while performing multiple tasks, such as point-to-point navigation, human-following, and -guiding. However, a prominent gap persists: in Human-Robot Interaction (HRI), the procedure of communicating commands to robots demands intricate mathematical formulations. Furthermore, the transition between tasks does not quite possess the intuitive control and user-centric interactivity that one would desire. In this work, we propose an LLM-driven interactive multimodal multitask robot navigation framework, termed LIM2N, to solve the above new challenge in the navigation field. We achieve this by first introducing a multimodal interaction framework where language and hand-drawn inputs can serve as navigation constraints and control objectives. Next, a reinforcement learning agent is built to handle multiple tasks with the received information. Crucially, LIM2N creates smooth cooperation among the reasoning of multimodal input, multitask planning, and adaptation and processing of the intelligent sensing modules in the complicated system. Detailed experiments are conducted in both simulation and the real world demonstrating that LIM2N has solid user needs understanding, alongside an enhanced interactive experience. Weiqin Zu, Wenbin Song, Ruiqing Chen, Ze Guo, Fanglei Sun, Zheng Tian 0002, Wei Pan 0004, Jun Wang 0012 |
ICRA | 8 |
| 2024 | Policy Learning from Tutorial Books via Understanding, Rehearsing and IntrospectingabstractWhen humans need to learn a new skill, we can acquire knowledge through written books, including textbooks, tutorials, etc. However, current research for decision-making, like reinforcement learning (RL), has primarily required numerous real interactions with the target environment to learn a skill, while failing to utilize the existing knowledge already summarized in the text. The success of Large Language Models (LLMs) sheds light on utilizing such knowledge behind the books. In this paper, we discuss a new policy learning problem called Policy Learning from tutorial Books (PLfB) upon the shoulders of LLMs’ systems, which aims to leverage rich resources such as tutorial books to derive a policy network. Inspired by how humans learn from books, we solve the problem via a three-stage framework: Understanding, Rehearsing, and Introspecting (URI). In particular, it first rehearses decision-making trajectories based on the derived knowledge after understanding the books, then introspects in the imaginary dataset to distill a policy network.
We build two benchmarks for PLfB~based on Tic-Tac-Toe and Football games. In experiment, URI's policy achieves at least 44% net win rate against GPT-based agents without any real data; In Football game, which is a complex scenario, URI's policy beat the built-in AIs with a 37% while using GPT-based agent can only achieve a 6\% winning rate. The project page: https://plfb-football.github.io. Xiong-Hui Chen, Yali Du 0001, Shengyi Jiang, Yang Yu 0001, Jun Wang 0012 |
NeurIPS | 7 |
| 2024 | Large Language Models Play StarCraft II: Benchmarks and A Chain of Summarization ApproachabstractWith the continued advancement of Large Language Models (LLMs) Agents in reasoning, planning, and decision-making, benchmarks have become crucial in evaluating these skills. However, there is a notable gap in benchmarks for real-time strategic decision-making. StarCraft II (SC2), with its complex and dynamic nature, serves as an ideal setting for such evaluations. To this end, we have developed TextStarCraft II, a specialized environment for assessing LLMs in real-time strategic scenarios within SC2. Addressing the limitations of traditional Chain of Thought (CoT) methods, we introduce the Chain of Summarization (CoS) method, enhancing LLMs' capabilities in rapid and effective decision-making. Our key experiments included:
1. LLM Evaluation: Tested 10 LLMs in TextStarCraft II, most of them defeating LV5 build-in AI, showcasing effective strategy skills.
2. Commercial Model Knowledge: Evaluated four commercial models on SC2 knowledge; GPT-4 ranked highest by Grandmaster-level experts.
3. Human-AI Matches: Experimental results showed that fine-tuned LLMs performed on par with Gold-level players in real-time matches, demonstrating comparable strategic abilities.
All code and data from this
study have been made pulicly available at https://github.com/histmeisah/Large-Language-Models-play-StarCraftII Weiyu Ma, Qirui Mi, Yongcheng Zeng, Runji Lin, Yuqiao Wu, Jun Wang 0012, Haifeng Zhang 0002 |
NeurIPS | 7 |
| 2024 | Reinforcing LLM Agents via Policy Optimization with Action DecompositionabstractLanguage models as intelligent agents push the boundaries of sequential decision-making agents but struggle with limited knowledge of environmental dynamics and exponentially huge action space. Recent efforts like GLAM and TWOSOME manually constrain the action space to a restricted subset and employ reinforcement learning to align agents' knowledge with specific environments. However, they overlook fine-grained credit assignments for intra-action tokens, which is essential for efficient language agent optimization, and rely on human's prior knowledge to restrict action space. This paper proposes decomposing language agent optimization from the action level to the token level, offering finer supervision for each intra-action token and manageable optimization complexity in environments with unrestricted action spaces. Beginning with the simplification of flattening all actions, we theoretically explore the discrepancies between action-level optimization and this naive token-level optimization. We then derive the Bellman backup with Action Decomposition (BAD) to integrate credit assignments for both intra-action and inter-action tokens, effectively eliminating the discrepancies. Implementing BAD within the PPO algorithm, we introduce Policy Optimization with Action Decomposition (POAD). POAD benefits from a finer-grained credit assignment process and lower optimization complexity, leading to enhanced learning efficiency and generalization abilities in aligning language agents with interactive environments. We validate POAD across diverse testbeds, with results affirming the advantages of our approach and the correctness of our theoretical analysis. The source code can be accessed directly with this link: https://github.com/morning9393/ADRL. Muning Wen, Ziyu Wan, Jun Wang 0012, Weinan Zhang 0001, Ying Wen 0001 |
NeurIPS | 3 |
| 2024 | InfoRank: Unbiased Learning-to-Rank via Conditional Mutual Information MinimizationabstractRanking items regarding individual user interests is a core technique of multiple downstream tasks such as recommender systems. Learning such a personalized ranker typically relies on the implicit feedback from users' past click-through behaviors. However, collected feedback is biased toward previously highly-ranked items and directly learning from it would result in "rich-get-richer" phenomena. In this paper, we propose a simple yet sufficient unbiased learning-to-rank paradigm named InfoRank that aims to simultaneously address both position and popularity biases. We begin by consolidating the impacts of those biases into a single observation factor, thereby providing a unified approach to addressing bias-related issues. Subsequently, we minimize the mutual information between the observation estimation and the relevance estimation conditioned on the input features. By doing so, our relevance estimation can be proved to be free of bias. To implement InfoRank, we first incorporate an attention mechanism to capture latent correlations within user-item features, thereby generating estimations of observation and relevance. We then introduce a regularization term, grounded in conditional mutual information, to promote conditional independence between relevance estimation and observation estimation. Experimental evaluations conducted across three extensive recommendation and search datasets reveal that InfoRank learns more precise and unbiased ranking strategies. Jiarui Jin, Zexue He, Mengyue Yang, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012, Julian J. McAuley |
WWW | 6 |
| 2024 | Manas: multi-agent neural architecture search
Vasco Lopes, Fabio Maria Carlucci, Pedro M. Esperança, Marco Singh, Antoine Yang, Victor Gabillon, Zewei Chen, Jun Wang 0012 |
Mach. Learn. | 9 |
| 2024 | A Review of Safe Reinforcement Learning: Methods, Theories, and ApplicationsabstractReinforcement Learning (RL) has achieved tremendous success in many complex decision-making tasks. However, safety concerns are raised during deploying RL in real-world applications, leading to a growing demand for safe RL algorithms, such as in autonomous driving and robotics scenarios. While safe control has a long history, the study of safe RL algorithms is still in the early stages. To establish a good foundation for future safe RL research, in this paper, we provide a review of safe RL from the perspectives of methods, theories, and applications. First, we review the progress of safe RL from five dimensions and come up with five crucial problems for safe RL being deployed in real-world applications, coined as "2H3W". Second, we analyze the algorithm and theory progress from the perspectives of answering the "2H3W" problems. Particularly, the sample complexity of safe RL algorithms is reviewed and discussed, followed by an introduction to the applications and benchmarks of safe RL algorithms. Finally, we open the discussion of the challenging problems in safe RL, hoping to inspire future research on this thread. To advance the study of safe RL algorithms, we release an open-sourced repository containing major safe RL algorithms at the link. Shangding Gu, Long Yang 0004, Yali Du 0001, Guang Chen 0001, Florian Walter, Jun Wang 0012, Alois C. Knoll |
IEEE Trans. Pattern Anal. Mach. Intell. | 6 |
| 2024 | Fear-Neuro-Inspired Reinforcement Learning for Safe Autonomous DrivingabstractEnsuring safety and achieving human-level driving performance remain challenges for autonomous vehicles, especially in safety-critical situations. As a key component of artificial intelligence, reinforcement learning is promising and has shown great potential in many complex tasks; however, its lack of safety guarantees limits its real-world applicability. Hence, further advancing reinforcement learning, especially from the safety perspective, is of great importance for autonomous driving. As revealed by cognitive neuroscientists, the amygdala of the brain can elicit defensive responses against threats or hazards, which is crucial for survival in and adaptation to risky environments. Drawing inspiration from this scientific discovery, we present a fear-neuro-inspired reinforcement learning framework to realize safe autonomous driving through modeling the amygdala functionality. This new technique facilitates an agent to learn defensive behaviors and achieve safe decision making with fewer safety violations. Through experimental tests, we show that the proposed approach enables the autonomous driving agent to attain state-of-the-art performance compared to the baseline agents and perform comparably to 30 certified human drivers, across various safety-critical scenarios. The results demonstrate the feasibility and effectiveness of our framework while also shedding light on the crucial role of simulating the amygdala function in the application of reinforcement learning to safety-critical autonomous driving domains. Xiangkun He, Jingda Wu, Zhiyu Huang, Zhongxu Hu, Jun Wang 0012, Alberto L. Sangiovanni-Vincentelli, Chen Lv 0001 |
IEEE Trans. Pattern Anal. Mach. Intell. | 5 |
| 2024 | Cross-Utterance Conditioned VAE for Speech GenerationabstractSpeech synthesis systems powered by neural networks hold promise for multimedia production, but frequently face issues with producing expressive speech and seamless editing. In response, we present the Cross-Utterance Conditioned Variational Autoencoder speech synthesis (CUC-VAE S2) framework to enhance prosody and ensure natural speech generation. This framework leverages the powerful representational capabilities of pre-trained language models and the re-expression abilities of variational autoencoders (VAEs). The core component of the CUC-VAE S2 framework is the cross-utterance CVAE, which extracts acoustic, speaker, and textual features from surrounding sentences to generate context-sensitive prosodic features, more accurately emulating human prosody generation. We further propose two practical algorithms tailored for distinct speech synthesis applications: CUC-VAE TTS for text-to-speech and CUC-VAE SE for speech editing. The CUC-VAE TTS is a direct application of the framework, designed to generate audio with contextual prosody derived from surrounding texts. On the other hand, the CUC-VAE SE algorithm leverages real mel spectrogram sampling conditioned on contextual information, producing audio that closely mirrors real sound and thereby facilitating flexible speech editing based on text such as deletion, insertion, and replacement. Experimental results on the LibriTTS datasets demonstrate that our proposed models significantly enhance speech synthesis and editing, producing more natural and expressive speech. Yang Li 0116, Guangzhi Sun, Weiqin Zu, Zheng Tian 0002, Ying Wen 0001, Wei Pan 0004, Chao Zhang 0031, Jun Wang 0012, Yang Yang 0001, Fanglei Sun |
IEEE ACM Trans. Audio Speech Lang. Process. | 9 |
| 2024 | Self-Supervised MAFENN for Classifying Low-Labeled Distorted Images Over Mobile Fading ChannelsabstractImage distortion during wireless transmission presents a significant challenge for real-world artificial intelligence (AI) applications. Recent methods have attempted to address this issue by integrating neural networks into the wireless transmission system. However, these approaches often require a large volume of labeled training data, which can be expensive and time-consuming to collect. To address this issue, we propose a novel approach,Self-SupervisedMulti-AgentFeedbackEnabledNeuralNetworks (S2MAFENN). S2MAFENN is designed to improve the efficiency of labeled data in wireless image transmission. It incorporates a Feedbacker agent that emulates the error correction mechanisms observed in primate brains and employs self-supervised contrastive learning to extract representations from unlabeled distorted images independently. From a theoretical perspective, we model the training process of S2MAFENN as a three-player Stackelberg game and provide evidence that S2MAFENN can achieve exponential convergence rates. We then empirically validate our approach by assessing the representations learned through S2MAFENN. We use varied labeled CIFAR10 and CIFAR100 data to simulate real image transmissions over the Rayleigh fading and 5G channels. Our results show that S2MAFENN matches or even surpasses the performance of state-of-the-art self-supervised training methods, even when only 50% of labels are used. Moreover, S2MAFENN yields average accuracy gains of 5.11%, 5.8%, and 4.58% with only 0.1, 0.2, and 0.5 of the labels transmitted over the 5G channel, respectively. For the downstream task of semantic segmentation over the 5G channel, S2MAFENN exhibits significant advancements on the ADE20K dataset. It achieves enhancements of approximately 7% and 8.7% in Mean IoU and DICE metrics, respectively, surpassing the performance of current state-of-the-art methods. Yang Li 0116, Fanglei Sun, Jingchen Hu, Fan Wu 0006, Kai Li 0022, Ying Wen 0001, Zheng Tian 0002, Yaodong Yang 0001, Jiangcheng Zhu, Jun Wang 0012, Yang Yang 0001 |
IEEE Trans. Mob. Comput. | 12 |
| 2023 | Learning to Shape Rewards Using a Game of Two PartnersabstractReward shaping (RS) is a powerful method in reinforcement learning (RL) for overcoming the problem of sparse or uninformative rewards. However, RS typically relies on manually engineered shaping-reward functions whose construc- tion is time-consuming and error-prone. It also requires domain knowledge which runs contrary to the goal of autonomous learning. We introduce Reinforcement Learning Optimising Shaping Algorithm (ROSA), an automated reward shaping framework in which the shaping-reward function is constructed in a Markov game between two agents. A reward-shaping agent (Shaper) uses switching controls to determine which states to add shaping rewards for more efficient learning while the other agent (Controller) learns the optimal policy for the task using these shaped rewards. We prove that ROSA, which adopts existing RL algorithms, learns to construct a shaping-reward function that is beneficial to the task thus ensuring efficient convergence to high performance policies. We demonstrate ROSA’s properties in three didactic experiments and show its superior performance against state-of-the-art RS algorithms in challenging sparse reward environments. David Mguni, Taher Jafferjee, Nicolas Perez Nieves, Wenbin Song, Feifei Tong, Matthew E. Taylor, Tianpei Yang, Zipeng Dai, Jiangcheng Zhu, Kun Shao, Jun Wang 0012, Yaodong Yang 0001 |
AAAI | 13 |
| 2023 | Replace Scoring with Arrangement: A Contextual Set-to-Arrangement Framework for Learning-to-RankabstractLearning-to-rank is a core technique in the top-N recommendation task, where an ideal ranker would be a mapping from an item set to an arrangement (a.k.a. permutation). Most existing solutions fall in the paradigm of probabilistic ranking principle (PRP), i.e., first score each item in the candidate set and then perform a sort operation to generate the top ranking list. However, these approaches neglect the contextual dependence among candidate items during individual scoring, and the sort operation is non-differentiable. To bypass the above issues, we propose Set-To-Arrangement Ranking (STARank), a new framework directly generates the permutations of the candidate items without the need for individually scoring and sort operations; and is end-to-end differentiable. As a result, STARank can operate when only the ground-truth permutations are accessible without requiring access to the ground-truth relevance scores for items. For this purpose, STARank first reads the candidate items in the context of the user browsing history, whose representations are fed into a Plackett-Luce module to arrange the given items into a list. To effectively utilize the given ground-truth permutations for supervising STARank, we leverage the internal consistency property of Plackett-Luce models to derive a computationally efficient list-wise loss. Experimental comparisons against 9 the state-of-the-art methods on 2 learning-to-rank benchmark datasets and 3 top-N real-world recommendation datasets demonstrate the superiority of STARank in terms of conventional ranking metrics. Notice that these ranking metrics do not consider the effects of the contextual dependence among the items in the list, we design a new family of simulation-based ranking metrics, where existing metrics can be regarded as special cases. STARank can consistently achieve better performance in terms of PBM and UBM simulation-based metrics. Jiarui Jin, Weinan Zhang 0001, Mengyue Yang, Yang Wang 0019, Yali Du 0001, Yong Yu 0001, Jun Wang 0012 |
CIKM | 8 |
| 2023 | How Do Large Language Models Capture the Ever-changing World Knowledge? A Review of Recent AdvancesabstractAlthough large language models (LLMs) are impressive in solving various tasks, they can quickly be outdated after deployment.Maintaining their up-to-date status is a pressing concern in the current era.This paper provides a comprehensive review of recent advances in aligning LLMs with the ever-changing world knowledge without re-training from scratch.We categorize research works systemically and provide in-depth comparisons and discussion.We also discuss existing challenges and highlight future directions to facilitate research in this field 1 . Ling Chen 0006, Mohammad-Reza Namazi-Rad, Jun Wang 0012 |
EMNLP | 5 |
| 2023 | PPO-Based Energy-Efficient Power Control and Spectrum Allocation in In-Vehicle HetNetsabstractWith the rapid development of intelligent vehicles in recent years, in-vehicle heterogeneous networks (HetNets) incorporating base stations (BSs) and vehicle access points (VAPs) have been widely deployed to support the ever-emerging diverse vehicular applications. However, most of the existing works focused on the HetNets covering multiple vehicles and considered the needs of all in-vehicle users as a holistic entity to maximize the overall performance of networks, which inevitably deviates from the local in-vehicle HetNet quality that most users are concerned about. Additionally, improving the energy efficiency (EE) of mobile devices in vehicles to extend their battery life is another significant issue, which has not been well addressed. To address the above issues, an intelligent in-vehicle power control and spectrum allocation mechanism is proposed in this work to maximize the EE of devices in the cabin while satisfying their dynamic traffic demands. Since this optimization problem has a non-convex mixed integer programming form, which is difficult to solve with traditional optimization methods, we further transform the optimization problem into a Markov Decision Process (MDP) and utilize a Proximal Policy Optimization (PPO) algorithm combined with the vehicle's location and historical channel state information (CSI) to achieve optimization objectives. Simulation results validate that the proposed algorithm can satisfy the dynamic traffic requirements of devices with high EE. Further comparison with baselines highlights the robustness of the proposed scheme under different device quantities and ratios. Tianyi Lin, Jun Du 0001, Haijun Zhang 0001, Arumugam Nallanathan, Jun Wang 0012 |
GLOBECOM | 5 |
| 2023 | Enhanced Discrete Multi-modal Hashing: More Constraints yet Less Time to Learn (Extended Abstract)abstractThis paper proposes a novel method, Enhanced Discrete Multi-modal Hashing (EDMH), which learns binary codes and hash functions simultaneously from the pairwise similarity matrix of data for large-scale cross-view retrieval. EDMH distinguishes itself from existing methods by considering not just the binarization constraint but also the balance and decorrelation constraints. Although those additional discrete constraints make the optimization problem of EDMH look a lot more complicated, we are actually able to develop a fast iterative learning algorithm in the alternating optimization framework for it, as after introducing a couple of auxiliary variables each subproblem of optimization turns out to have closed-form solutions. It has been confirmed by extensive experiments that EDMH can consistently deliver better retrieval performances than state-of-the-art MH methods at lower computational costs. Yong Chen 0008, Hui Zhang 0028, Zhibao Tian, Jun Wang 0012, Dell Zhang, Xuelong Li 0001 |
ICDE | 4 |
| 2023 | CounterCLR: Counterfactual Contrastive Learning with Non-random Missing Data in RecommendationabstractRecommender systems are designed to learn user preferences from observed feedback and comprise many fundamental tasks, such as rating prediction and post-click conversion rate (pCVR) prediction. However, the observed feedback usually suffer from two issues: selection bias and data sparsity, where biased and insufficient feedback seriously degrade the performance of recommender systems in terms of accuracy and ranking. Existing solutions for handling the issues, such as data imputation and inverse propensity score, are highly susceptible to additional trained imputation or propensity models. In this work, we propose a novel counterfactual contrastive learning framework for recommendation, named CounterCLR, to tackle the problem of non-random missing data by exploiting the advances in contrast learning. Specifically, the proposed CounterCLR employs a deep representation network, called CauNet, to infer non-random missing data in recommendations and perform user preference modeling by further introducing a self-supervised contrastive learning task. Our CounterCLR mitigates the selection bias problem without the need for additional models or estimators, while also enhancing the generalization ability in cases of sparse data. Experiments on real-world datasets demonstrate the effectiveness and superiority of our method. Jun Wang 0012, Haoxuan Li 0001, Dongxu Liang, Enyun Yu, Wenwu Ou |
ICDM | 1 |
| 2023 | Timing is Everything: Learning to Act Selectively with Costly Actions and Budgetary Constraints
David Mguni, Aivar Sootla, Juliusz Krysztof Ziomek, Oliver Slumbers, Zipeng Dai, Kun Shao, Jun Wang 0012 |
ICLR | 7 |
| 2023 | Order Matters: Agent-by-agent Policy Optimization
Xihuai Wang, Zheng Tian 0002, Ziyu Wan, Ying Wen 0001, Jun Wang 0012, Weinan Zhang 0001 |
ICLR | 5 |
| 2023 | MANSA: Learning Fast and Slow in Multi-Agent SystemsabstractIn multi-agent reinforcement learning (MARL), independent learning (IL) often shows remarkable performance and easily scales with the number of agents. Yet, using IL can be inefficient and runs the risk of failing to successfully train, particularly in scenarios that require agents to coordinate their actions. Using centralised learning (CL) enables MARL agents to quickly learn how to coordinate their behaviour but employing CL everywhere is often prohibitively expensive in real-world applications. Besides, using CL in value-based methods often needs strong representational constraints (e.g. individual-global-max condition) that can lead to poor performance if violated. In this paper, we introduce a novel plug & play IL framework named Multi-Agent Network Selection Algorithm (MANSA) which selectively employs CL only at states that require coordination. At its core, MANSA has an additional agent that uses switching controls to quickly learn the best states to activate CL during training, using CL only where necessary and vastly reducing the computational burden of CL. Our theory proves MANSA preserves cooperative MARL convergence properties, boosts IL performance and can optimally make use of a fixed budget on the number CL calls. We show empirically in Level-based Foraging (LBF) and StarCraft Multi-agent Challenge (SMAC) that MANSA achieves fast, superior and more reliable performance while making 40% fewer CL calls in SMAC and using CL at only 1% CL calls in LBF. David Mguni, Haojun Chen, Taher Jafferjee, Longfei Yue, Xidong Feng, Stephen McAleer, Feifei Tong, Jun Wang 0012, Yaodong Yang 0001 |
ICML | 9 |
| 2023 | A Game-Theoretic Framework for Managing Risk in Multi-Agent SystemsabstractIn order for agents in multi-agent systems (MAS) to be safe, they need to take into account the risks posed by the actions of other agents. However, the dominant paradigm in game theory (GT) assumes that agents are not affected by risk from other agents and only strive to maximise their expected utility. For example, in hybrid human-AI driving systems, it is necessary to limit large deviations in reward resulting from car crashes. Although there are equilibrium concepts in game theory that take into account risk aversion, they either assume that agents are risk-neutral with respect to the uncertainty caused by the actions of other agents, or they are not guaranteed to exist. We introduce a new GT-based Risk-Averse Equilibrium (RAE) that always produces a solution that minimises the potential variance in reward accounting for the strategy of other agents. Theoretically and empirically, we show RAE shares many properties with a Nash Equilibrium (NE), establishing convergence properties and generalising to risk-dominant NE in certain cases. To tackle large-scale problems, we extend RAE to the PSRO multi-agent reinforcement learning (MARL) framework. We empirically demonstrate the minimum reward variance benefits of RAE in matrix games with high-risk outcomes. Results on MARL experiments show RAE generalises to risk-dominant NE in a trust dilemma game and that it reduces instances of crashing by 7x in an autonomous driving setting versus the best performing baseline. Oliver Slumbers, David Mguni, Stefano B. Blumberg, Stephen McAleer, Yaodong Yang 0001, Jun Wang 0012 |
ICML | 6 |
| 2023 | GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning ModelsabstractThis paper introduces a distributed, GPU-centric experience replay system, GEAR, designed to perform scalable reinforcement learning (RL) with large sequence models (such as transformers). With such models, existing systems such as Reverb face considerable bottlenecks in memory, computation, and communication. GEAR, however, optimizes memory efficiency by enabling the memory resources on GPU servers (including host memory and device memory) to manage trajectory data. Furthermore, it facilitates decentralized GPU devices to expedite various trajectory selection strategies, circumventing computational bottlenecks. GEAR is equipped with GPU kernels capable of collecting trajectories using zero-copy access to host memory, along with remote-directed-memory access over InfiniBand, improving communication efficiency. Cluster experiments have shown that GEAR can achieve performance levels up to 6× greater than Reverb when training state-of-the-art large RL models. GEAR is open-sourced at https:// github.com/bigrl-team/gear. Hanjing Wang, Man-Kit Sit, Congjie He, Ying Wen 0001, Weinan Zhang 0001, Jun Wang 0012, Yaodong Yang 0001, Luo Mai |
ICML | 6 |
| 2023 | Reinforcement Learning for Safe Robot Control using Control Lyapunov Barrier FunctionsabstractReinforcement learning (RL) exhibits impressive performance when managing complicated control tasks for robots. However, its wide application to physical robots is limited by the absence of strong safety guarantees. To overcome this challenge, this paper explores the control Lyapunov barrier function (CLBF) to analyze the safety and reachability solely based on data without explicitly employing a dynamic model. We also proposed the Lyapunov barrier actor-critic (LBAC), a model-free RL algorithm, to search for a controller that satisfies the data-based approximation of the safety and reachability conditions. The proposed approach is demonstrated through simulation and real-world robot control experiments, i.e., a 2D quadrotor navigation task. The experimental findings reveal this approach's effectiveness in reachability and safety, surpassing other model-free RL methods. Desong Du, Shaohang Han, Naiming Qi, Haitham Bou-Ammar, Jun Wang 0012, Wei Pan 0004 |
ICRA | 5 |
| 2023 | Sim-to-Real Transfer for Quadrupedal Locomotion via Terrain TransformerabstractDeep reinforcement learning has recently emerged as an appealing alternative for legged locomotion over multiple terrains by training a policy in physical simulation and then transferring it to the real world (i.e., sim-to-real transfer). Despite considerable progress, the capacity and scalability of traditional neural networks are still limited, which may hinder their applications in more complex environments. In contrast, the Transformer architecture has shown its superiority in a wide range of large-scale sequence modeling tasks, including natural language processing and decision-making problems. In this paper, we propose Terrain Transformer (TERT), a high-capacity Transformer model for quadrupedal locomotion control on various terrains. Furthermore, to better leverage Transformer in sim-to-real scenarios, we present a novel two-stage training framework consisting of an offline pretraining stage and an online correction stage, which can naturally integrate Transformer with privileged training. Extensive experiments in simulation demonstrate that TERT outperforms state-of-the-art baselines on different terrains in terms of return, energy consumption and control smoothness. In further real-world validation, TERT successfully traverses nine challenging terrains, including sand pit and stair down, which can not be accomplished by strong baselines. Hang Lai, Weinan Zhang 0001, Xialin He, Zheng Tian 0002, Yong Yu 0001, Jun Wang 0012 |
ICRA | 7 |
| 2023 | Multi-embodiment Legged Robot Control as a Sequence Modeling ProblemabstractRobots are traditionally bounded by a fixed embodiment during their operational lifetime, which limits their ability to adapt to their surroundings. Co-optimizing control and morphology of a robot, however, is often inefficient due to the complex interplay between the controller and morphology. In this paper, we propose a learning-based control method that can inherently take morphology into consideration such that once the control policy is trained in the simulator, it can be easily deployed to real robots with different embodiments. In particular, we present the Embodiment-aware Transformer (EAT), an architecture that casts this control problem as conditional sequence modeling. EAT outputs the optimal actions by leveraging a causally masked Transformer. By conditioning an autoregressive model on the desired robot embodiment, past states, and actions, our EAT model can generate future actions that best fit the current robot embodiment. Experimental results show that EAT can outperform all other alternatives in embodiment-varying tasks, and succeed in an example of real-world evolution tasks: stepping down a stair through updating the morphology alone. We hope that EAT will inspire a new push toward real-world evolution across many domains, where algorithms like EAT can blaze a trail by bridging the field of evolutionary robotics and big data sequence modeling. Weinan Zhang 0001, Hang Lai, Zheng Tian 0002, Laurent Kneip, Jun Wang 0012 |
ICRA | 6 |
| 2023 | Specify Robust Causal Representation from Mixed ObservationsabstractLearning representations purely from observations concerns the problem of learning a low-dimensional, compact representation which is beneficial to prediction models. Under the hypothesis that the intrinsic latent factors follow some casual generative models, we argue that by learning a causal representation, which is the minimal sufficient causes of the whole system, we can improve the robustness and generalization performance of machine learning models. In this paper, we develop a learning method to learn such representation from observational data by regularizing the learning procedure with mutual information measures, according to the hypothetical factored causal graph. We theoretically and empirically show that the models trained with the learned causal representations are more robust under adversarial attacks and distribution shifts compared with baselines. Mengyue Yang, Xinyu Cai, Furui Liu, Weinan Zhang 0001, Jun Wang 0012 |
KDD | 5 |
| 2023 | Online PCA in Converging Self-consistent Field EquationsabstractSelf-consistent Field (SCF) equation is a type of nonlinear eigenvalue problem in which the matrix to be eigen-decomposed is a function of its own eigenvectors. It is of great significance in computational science for its connection to the Schrödinger equation. Traditional fixed-point iteration methods for solving such equations suffer from non-convergence issues. In this work, we present a novel perspective on such SCF equations as a principal component analysis (PCA) for non-stationary time series, in which a distribution and its own top principal components are mutually updated over time, and the equilibrium state of the model corresponds to the solution of the SCF equations. By the new perspective, online PCA techniques are able to engage in so as to enhance the convergence of the model towards the equilibrium state, acting as a new set of tools for converging the SCF equations. With several numerical adaptations, we then develop a new algorithm for converging the SCF equation, and demonstrated its high convergence capacity with experiments on both synthesized and real electronic structure scenarios. Xihan Li 0001, Rasul Tutunov, Haitham Bou-Ammar, Lei Wang 0198, Jun Wang 0012 |
NeurIPS | 6 |
| 2023 | Interpretable Reward Redistribution in Reinforcement Learning: A Causal ApproachabstractA major challenge in reinforcement learning is to determine which state-action pairs are responsible for future rewards that are delayed. Reward redistribution serves as a solution to re-assign credits for each time step from observed sequences. While the majority of current approaches construct the reward redistribution in an uninterpretable manner, we propose to explicitly model the contributions of state and action from a causal perspective, resulting in an interpretable reward redistribution and preserving policy invariance. In this paper, we start by studying the role of causal generative models in reward redistribution by characterizing the generation of Markovian rewards and trajectory-wise long-term return and further propose a framework, called Generative Return Decomposition (GRD), for policy optimization in delayed reward scenarios. Specifically, GRD first identifies the unobservable Markovian rewards and causal relations in the generative process. Then, GRD makes use of the identified causal generative model to form a compact representation to train policy over the most favorable subspace of the state space of the agent. Theoretically, we show that the unobservable Markovian reward function is identifiable, as well as the underlying causal structure and causal models. Experimental results show that our method outperforms state-of-the-art methods and the provided visualization further demonstrates the interpretability of our method.
The project page is located at [https://reedzyd.github.io/GenerativeReturnDecomposition/](https://reedzyd.github.io/GenerativeReturnDecomposition/). Yudi Zhang 0006, Yali Du 0001, Biwei Huang, Jun Wang 0012, Mykola Pechenizkiy |
NeurIPS | 5 |
| 2023 | ChessGPT: Bridging Policy Learning and Language ModelingabstractWhen solving decision-making tasks, humans typically depend on information from two key sources: (1) Historical policy data, which provides interaction replay from the environment, and (2) Analytical insights in natural language form, exposing the invaluable thought process or strategic considerations. Despite this, the majority of preceding research focuses on only one source: they either use historical replay exclusively to directly learn policy or value functions, or engaged in language model training utilizing mere language corpus. In this paper, we argue that a powerful autonomous agent should cover both sources. Thus, we propose ChessGPT, a GPT model bridging policy learning and language modeling by integrating data from these two sources in Chess games. Specifically, we build a large-scale game and language dataset related to chess. Leveraging the dataset, we showcase two model examples ChessCLIP and ChessGPT, integrating policy learning and language modeling. Finally, we propose a full evaluation framework for evaluating language model's chess ability. Experimental results validate our model and dataset's effectiveness. We open source our code, model, and dataset at https://github.com/waterhorse1/ChessGPT. Xidong Feng, Yicheng Luo, Hongrui Tang, Mengyue Yang, Kun Shao, David Mguni, Yali Du 0001, Jun Wang 0012 |
NeurIPS | 9 |
| 2023 | Lending Interaction Wings to Recommender Systems with Conversational AgentsabstractAn intelligent conversational agent (a.k.a., chat-bot) could embrace conversational technologies to obtain user preferences online, to overcome inherent limitations of recommender systems trained over the offline historical user behaviors. In this paper, we propose CORE, a new offline-training and online-checking framework to plug a COnversational agent into REcommender systems. Unlike most prior conversational recommendation approaches that systemically combine conversational and recommender parts through a reinforcement learning framework, CORE bridges the conversational agent and recommender system through a unified uncertainty minimization framework, which can be easily applied to any existing recommendation approach. Concretely, CORE treats a recommender system as an offline estimator to produce an estimated relevance score for each item, while CORE regards a conversational agent as an online checker that checks these estimated scores in each online session. We define uncertainty as the sum of unchecked relevance scores. In this regard, the conversational agent acts to minimize uncertainty via querying either attributes or items. Towards uncertainty minimization, we derive the certainty gain of querying each attribute and item, and develop a novel online decision tree algorithm to decide what to query at each turn. Our theoretical analysis reveals the bound of the expected number of turns of CORE in a cold-start setting. Experimental results demonstrate that CORE can be seamlessly employed on a variety of recommendation approaches, and can consistently bring significant improvements in both hot-start and cold-start settings. Jiarui Jin, Fanghua Ye 0001, Mengyue Yang, Yue Feng 0002, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
NeurIPS | 8 |
| 2023 | An Efficient End-to-End Training Approach for Zero-Shot Human-AI CoordinationabstractThe goal of zero-shot human-AI coordination is to develop an agent that can collaborate with humans without relying on human data. Prevailing two-stage population-based methods require a diverse population of mutually distinct policies to simulate diverse human behaviors. The necessity of such populations severely limits their computational efficiency. To address this issue, we propose E3T, an **E**fficient **E**nd-to-**E**nd **T**raining approach for zero-shot human-AI coordination. E3T employs a mixture of ego policy and random policy to construct the partner policy, making it both coordination-skilled and diverse. In this way, the ego agent is end-to-end trained with this mixture policy without the need of a pre-trained population, thus significantly improving the training efficiency. In addition, a partner modeling module is proposed to predict the partner's action from historical information. With the predicted partner's action, the ego policy is able to adapt its policy and take actions accordingly when collaborating with humans of different behavior patterns. Empirical results on the Overcooked environment show that our method significantly improves the training efficiency while preserving comparable or superior performance than the population-based baselines. Demo videos are available at https://sites.google.com/view/e3t-overcooked. Jiaxian Guo, Xingzhou Lou, Jun Wang 0012, Haifeng Zhang 0002, Yali Du 0001 |
NeurIPS | 4 |
| 2023 | Invariant Learning via Probability of Sufficient and Necessary CausesabstractOut-of-distribution (OOD) generalization is indispensable for learning models in the wild, where testing distribution typically unknown and different from the training. Recent methods derived from causality have shown great potential in achieving OOD generalization.
However, existing methods mainly focus on the invariance property of causes, while largely overlooking the property of sufficiency and necessity conditions. Namely, a necessary but insufficient cause (feature) is invariant to distribution shift, yet it may not have required accuracy. By contrast, a sufficient yet unnecessary cause (feature) tends to fit specific data well but may have a risk of adapting to a new domain.
To capture the information of sufficient and necessary causes, we employ a classical concept, the probability of sufficiency and necessary causes (PNS), which indicates the probability of whether one is the necessary and sufficient cause.
To associate PNS with OOD generalization, we propose PNS risk and formulate an algorithm to learn representation with a high PNS value. We theoretically analyze and prove the generalizability of the PNS risk. Experiments on both synthetic and real-world benchmarks demonstrate the effectiveness of the proposed method. The detailed implementation can be found at the GitHub repository: https://github.com/ymy4323460/CaSN. Mengyue Yang, Yonggang Zhang 0003, Zhen Fang 0001, Yali Du 0001, Furui Liu, Jean-Francois Ton, Jun Wang 0012 |
NeurIPS | 8 |
| 2023 | Rectifying Unfairness in Recommendation Feedback LoopabstractThe issue of fairness in recommendation systems has recently become a matter of growing concern for both the academic and industrial sectors due to the potential for bias in machine learning models. One such bias is that of feedback loops, where the collection of data from an unfair online system hinders the accurate evaluation of the relevance scores between users and items. Given that recommendation systems often recommend popular content and vendors, the underlying relevance scores between users and items may not be accurately represented in the training data. Hence, this creates a feedback loop in which the user is not longer recommended based on their true relevance score but instead based on biased training data. To address this problem of feedback loops, we propose a two-stage representation learning framework, B-FAIR, aimed at rectifying the unfairness caused by biased historical data in recommendation systems. The framework disentangles the context data into sensitive and non-sensitive components using a variational autoencoder and then applies a novel Balanced Fairness Objective (BFO) to remove bias in the observational data when training a recommendation model. The efficacy of B-FAIR is demonstrated through experiments on both synthetic and real-world benchmarks, showing improved performance over state-of-the-art algorithms. Mengyue Yang, Jun Wang 0012, Jean-Francois Ton |
SIGIR | 2 |
| 2023 | Minimizing Age of Information for Mobile Edge Computing Systems: A Nested Index ApproachabstractExploiting the computational heterogeneity of mobile devices and edge nodes, mobile edge computation (MEC) provides an efficient approach to achieving real-time applications that are sensitive to information freshness, by offloading tasks from mobile devices to edge nodes. We use the metric Age-of-Information (AoI) to evaluate information freshness. An efficient solution to minimize the AoI for the MEC system with multiple users is non-trivial to obtain due to the random computing time. In this paper, we consider multiple users offloading tasks to heterogeneous edge servers in a MEC system. We first reformulate the problem as a Restless Multi-Arm-Bandit (RMAB) problem and establish a hierarchical Markov Decision Process (MDP) to characterize the updating of AoI for the MEC system. Based on the hierarchical MDP, we propose a nested index framework and design a nested index policy with provably asymptotic optimality. Finally, the closed form of the nested index is obtained, which enables the performance tradeoffs between computation complexity and accuracy. Our algorithm leads to an optimality gap reduction of up to 40%, compared to benchmarks. Our algorithm asymptotically approximates the lower bound as the system scalar gets large enough. Ning Yang 0005, Meng Zhang 0013, Jun Wang 0012 |
WiOpt | 4 |
| 2023 | Online Markov decision processes with non-oblivious strategic adversary
Le Cong Dinh, David Mguni, Long Tran-Thanh, Jun Wang 0012, Yaodong Yang 0001 |
Auton. Agents Multi Agent Syst. | 4 |
| 2023 | Large sequence models for sequential decision-making: a survey
Muning Wen, Runji Lin, Hanjing Wang, Yaodong Yang 0001, Ying Wen 0001, Luo Mai, Jun Wang 0012, Haifeng Zhang 0002, Weinan Zhang 0001 |
Frontiers Comput. Sci. | 7 |
| 2023 | MALib: A Parallel Framework for Population-based Multi-agent Reinforcement LearningabstractPopulation-based multi-agent reinforcement learning (PB-MARL) encompasses a range of methods that merge dynamic population selection with multi-agent reinforcement learning algorithms (MARL). While PB-MARL has demonstrated notable achievements in complex multi-agent tasks, its sequential execution is plagued by low computational efficiency due to the diversity in computing patterns and policy combinations. We propose a solution involving a stateless central task dispatcher and stateful workers to handle PB-MARL's subroutines, thereby capitalizing on parallelism across various components for efficient problem-solving. In line with this approach, we introduce MALib, a parallel framework that incorporates a task control model, independent data servers, and an abstraction of MARL training paradigms. The framework has undergone extensive testing and is available under the MIT license (https://github.com/sjtu-marl/malib) Ming Zhou 0006, Ziyu Wan, Hanjing Wang, Muning Wen, Runzhe Wu, Ying Wen 0001, Yaodong Yang 0001, Yong Yu 0001, Jun Wang 0012, Weinan Zhang 0001 |
J. Mach. Learn. Res. | 9 |
| 2023 | GEO: A Computational Design Framework for Automotive Exterior FaceliftabstractExterior facelift has become an effective method for automakers to boost the consumers’ interest in an existing car model before it is redesigned. To support the automotive facelift design process, this study develops a novel computational framework – Generator, Evaluator, Optimiser (GEO) , which comprises three components: a StyleGAN2-based design generator that creates different facelift designs; a convolutional neural network (CNN) -based evaluator that assesses designs from the aesthetics perspective; and a recurrent neural network (RNN) -based decision optimiser that selects designs to maximise the predicted profit of the targeted car model over time. We validate the GEO framework in experiments with real-world datasets and describe some resulting managerial implications for automotive facelift. Our study makes both methodological and application contributions. First, the generator’s mapping network and projection methods are carefully tailored to facelift where only minor changes are performed without affecting the family signature of the automobile brands. Second, two evaluation metrics are proposed to assess the generated designs. Third, profit maximisation is taken into account in the design selection. From a high-level perspective, our study contributes to the recent use of machine learning and data mining in marketing and design studies. To the best of our knowledge, this is the first study that uses deep generative models for automotive regional design upgrading and that provides an end-to-end decision-support solution for automakers and designers. Jingmin Huang, Bowei Chen 0001, Zhi Yan 0001, Iadh Ounis, Jun Wang 0012 |
ACM Trans. Knowl. Discov. Data | 5 |
| 2023 | Debiased Recommendation with User Feature BalancingabstractDebiased recommendation has recently attracted increasing attention from both industry and academic communities. Traditional models mostly rely on the inverse propensity score (IPS), which can be hard to estimate and may suffer from the high variance issue. To alleviate these problems, in this article, we propose a novel debiased recommendation framework based on user feature balancing. The general idea is to introduce a projection function to adjust user feature distributions, such that the ideal unbiased learning objective can be upper bounded by a solvable objective purely based on the offline dataset. In the upper bound, the projected user distributions are expected to be equal given different items. From the causal inference perspective, this requirement aims to remove the causal relation from the user to the item, which enables us to achieve unbiased recommendation, bypassing the computation of IPS. To efficiently balance the user distributions upon each item pair, we propose three strategies, including clipping, sampling, and adversarial learning to improve the training process. For more robust optimization, we deploy an explicit model to capture the potential latent confounders in recommendation systems. To the best of our knowledge, this article is the first work on debiased recommendation based on confounder balancing. In the experiments, we compare our framework with many state-of-the-art methods based on synthetic, semi-synthetic, and real-world datasets. Extensive experiments demonstrate that our model is effective in promoting the recommendation performance. Mengyue Yang, Guohao Cai, Furui Liu, Jiarui Jin, Zhenhua Dong, Xiuqiang He 0001, Jianye Hao, Weiqi Shao, Jun Wang 0012, Xu Chen 0017 |
ACM Trans. Inf. Syst. | 9 |
| 2022 | Learning to Identify Top Elo Ratings: A Dueling Bandits ApproachabstractThe Elo rating system is widely adopted to evaluate the skills of (chess) game and sports players. Recently it has been also integrated into machine learning algorithms in evaluating the performance of computerised AI agents. However, an accurate estimation of the Elo rating (for the top players) often requires many rounds of competitions, which can be expensive to carry out. In this paper, to minimize the number of comparisons and to improve the sample efficiency of the Elo evaluation (for top players), we propose an efficient online match scheduling algorithm. Specifically, we identify and match the top players through a dueling bandits framework and tailor the bandit algorithm to the gradient-based update of Elo. We show that it reduces the per-step memory and time complexity to constant, compared to the traditional likelihood maximization approaches requiring O(t) time. Our algorithm has a regret guarantee that is sublinear in the number of competition rounds and has been extended to the multidimensional Elo ratings for handling intransitive games. We empirically demonstrate that our method achieves superior convergence speed and time efficiency on a variety of gaming tasks. Yali Du 0001, Binxin Ru, Jun Wang 0012, Haifeng Zhang 0002, Xu Chen 0017 |
AAAI | 4 |
| 2022 | Cross-Utterance Conditioned VAE for Non-Autoregressive Text-to-SpeechabstractYang Li, Cheng Yu, Guangzhi Sun, Hua Jiang, Fanglei Sun, Weiqin Zu, Ying Wen, Yang Yang, Jun Wang. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Yang Li 0116, Guangzhi Sun, Fanglei Sun, Weiqin Zu, Ying Wen 0001, Yang Yang 0001, Jun Wang 0012 |
ACL (1) | 9 |
| 2022 | A Game-Theoretic Approach to Multi-agent Trust Region Optimization
Ying Wen 0001, Yaodong Yang 0001, Minne Li, Zheng Tian 0002, Xu Chen 0017, Jun Wang 0012 |
DAI | 7 |
| 2022 | BOiLS: Bayesian Optimisation for Logic SynthesisabstractOptimising the quality-of-results (QoR) of circuits during logic synthesis is a formidable challenge necessitating the exploration of exponentially sized search spaces. While expert-designed operations aid in uncovering effective sequences, the increase in complexity of logic circuits favours automated procedures. To enable efficient and scalable solvers, we propose BOiLS, the first algorithm adapting Bayesian optimisation to navigate the space of synthesis operations. BOiLS requires no human intervention and trades-off exploration versus exploitation through novel Gaussian process kernels and trust-region constrained acquisitions. In a set of experiments on EPFL benchmarks, we demonstrate BOiLS's superior performance compared to state-of-the-art in terms of both sample efficiency and QoR values. Antoine Grosnit, Cédric Malherbe, Rasul Tutunov, Xingchen Wan, Jun Wang 0012, Haitham Bou-Ammar |
DATE | 5 |
| 2022 | LPCSE: Neural Speech Enhancement through Linear Predictive CodingabstractThe increasingly stringent requirement on quality-of-experience in 5G/B5G communication systems has led to the emerging neural speech enhancement techniques, which however have been developed in isolation from the existing expert-rule based models of speech pronunciation and distortion, such as the classic Linear Predictive Coding (LPC) speech model because it is difficult to integrate the models with auto-differentiable machine learning frameworks. In this paper, to improve the efficiency of neural speech enhancement, we introduce an LPC-based speech enhancement (LPCSE) architecture, which leverages the strong inductive biases in the LPC speech model in conjunction with the expressive power of neural networks. Differentiable end-to-end learning is achieved in LPCSE via two novel blocks: a block that utilizes the expert rules to reduce the computational overhead when integrating the LPC speech model into neural networks, and a block that ensures the stability of the model and avoids exploding gradients in end-to-end training by mapping the Linear prediction coefficients to the filter poles. The experimental results show that LPCSE successfully restores the formants of the speeches distorted by transmission loss, and outperforms two existing neural speech enhancement methods of comparable neural network sizes in terms of the Perceptual evaluation of speech quality (PESQ) and Short-Time Objective Intelligibility (STOI) on the LJ Speech corpus. Yang Liu 0047, Na Tang, Xiaoli Chu, Yang Yang 0001, Jun Wang 0012 |
GLOBECOM | 5 |
| 2022 | Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning
Jakub Grudzien Kuba, Ruiqing Chen, Muning Wen, Ying Wen 0001, Fanglei Sun, Jun Wang 0012, Yaodong Yang 0001 |
ICLR | 6 |
| 2022 | LIGS: Learnable Intrinsic-Reward Generation Selection for Multi-Agent Learning
David Mguni, Taher Jafferjee, Nicolas Perez Nieves, Oliver Slumbers, Feifei Tong, Yang Li 0116, Jiangcheng Zhu, Yaodong Yang 0001, Jun Wang 0012 |
ICLR | 10 |
| 2022 | Reinforcement Learning in Presence of Discrete Markovian Context Evolution
Aivar Sootla, Taher Jafferjee, Junxiao Shen, Jun Wang 0012, Haitham Bou-Ammar |
ICLR | 5 |
| 2022 | Learning State Representations via Retracing in Reinforcement Learning
Changmin Yu, Dong Li 0016, Jianye Hao, Jun Wang 0012, Neil Burgess |
ICLR | 4 |
| 2022 | Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy OptimizationabstractRecent progress in state-only imitation learning extends the scope of applicability of imitation learning to real-world settings by relieving the need for observing expert actions. However, existing solutions only learn to extract a state-to-action mapping policy from the data, without considering how the expert plans to the target. This hinders the ability to leverage demonstrations and limits the flexibility of the policy. In this paper, we introduce Decoupled Policy Optimization (DePO), which explicitly decouples the policy as a high-level state planner and an inverse dynamics model. With embedded decoupled policy gradient and generative adversarial training, DePO enables knowledge transfer to different action spaces or state transition dynamics, and can generalize the planner to out-of-demonstration state regions. Our in-depth experimental analysis shows the effectiveness of DePO on learning a generalized target state planner while achieving the best imitation performance. We demonstrate the appealing usage of DePO for transferring across different tasks by pre-training, and the potential for co-training agents with various skills. Minghuan Liu, Zhengbang Zhu, Yuzheng Zhuang, Weinan Zhang 0001, Jianye Hao, Yong Yu 0001, Jun Wang 0012 |
ICML | 7 |
| 2022 | Saute RL: Almost Surely Safe Reinforcement Learning Using State AugmentationabstractSatisfying safety constraints almost surely (or with probability one) can be critical for the deployment of Reinforcement Learning (RL) in real-life applications. For example, plane landing and take-off should ideally occur with probability one. We address the problem by introducing Safety Augmented (Saute) Markov Decision Processes (MDPs), where the safety constraints are eliminated by augmenting them into the state-space and reshaping the objective. We show that Saute MDP satisfies the Bellman equation and moves us closer to solving Safe RL with constraints satisfied almost surely. We argue that Saute MDP allows viewing the Safe RL problem from a different perspective enabling new features. For instance, our approach has a plug-and-play nature, i.e., any RL algorithm can be "Sauteed”. Additionally, state augmentation allows for policy generalization across safety constraints. We finally show that Saute RL algorithms can outperform their state-of-the-art counterparts when constraint satisfaction is of high importance. Aivar Sootla, Alexander I. Cowen-Rivers, Taher Jafferjee, David Mguni, Jun Wang 0012, Haitham Bou-Ammar |
ICML | 6 |
| 2022 | Understanding Policy Gradient Algorithms: A Sensitivity-Based ApproachabstractThe REINFORCE algorithm \cite{williams1992simple} is popular in policy gradient (PG) for solving reinforcement learning (RL) problems. Meanwhile, the theoretical form of PG is from \cite{sutton1999policy}. Although both formulae prescribe PG, their precise connections are not yet illustrated. Recently, \citeauthor{nota2020policy} (\citeyear{nota2020policy}) have found that the ambiguity causes implementation errors. Motivated by the ambiguity and implementation incorrectness, we study PG from a perturbation perspective. In particular, we derive PG in a unified framework, precisely clarify the relation between PG implementation and theory, and echos back the findings by \citeauthor{nota2020policy}. Diving into factors contributing to empirical successes of the existing erroneous implementations, we find that small approximation error and the experience replay mechanism play critical roles. Shuang Wu 0005, Ling Shi 0001, Jun Wang 0012, Guangjian Tian |
ICML | 3 |
| 2022 | Promoting Quality and Diversity in Population-based Reinforcement Learning via Hierarchical Trajectory Space ExplorationabstractQuality Diversity (QD) algorithms in population-based reinforcement learning aim to optimize agents' returns and diversity among the population simultaneously. It is conducive to solving exploration problems in reinforcement learning and potentially getting multiple good and diverse strategies. However, previous methods typically define behavioral embedding in action space or outcome space, which neglect trajectory characteristics during the execution process. In this paper, we introduce a trajectory embedding model trained by Variational Autoencoder with similarity constraint to characterize trajectory features. Based on that, we propose a hierarchical trajectory-space exploration (HTSE) framework using Determinantal Point Processes (DPP) to generate high-quality and diverse solutions in the selection and mutation process. The experimental results show that our HTSE method effectively completes several simulated tasks, outperforming other Quality-Diversity Reinforcement Learning algorithms. Jiayu Miao, Tianze Zhou, Kun Shao, Ming Zhou 0006, Weinan Zhang 0001, Jianye Hao, Yong Yu 0001, Jun Wang 0012 |
ICRA | 8 |
| 2022 | On the Convergence of Fictitious Play: A Decomposition ApproachabstractFictitious play (FP) is one of the most fundamental game-theoretical learning frameworks for computing Nash equilibrium in n-player games, which builds the foundation for modern multi-agent learning algorithms. Although FP has provable convergence guarantees on zero-sum games and potential games, many real-world problems are often a mixture of both and the convergence property of FP has not been fully studied yet. In this paper, we extend the convergence results of FP to the combinations of such games and beyond. Specifically, we derive new conditions for FP to converge by leveraging game decomposition techniques. We further develop a linear relationship unifying cooperation and competition in the sense that these two classes of games are mutually transferable. Finally, we analyse a non-convergent example of FP, the Shapley game, and develop sufficient conditions for FP to converge. Yurong Chen 0002, Xiaotie Deng, David Mguni, Jun Wang 0012, Yaodong Yang 0001 |
IJCAI | 5 |
| 2022 | Scalable Model-based Policy Optimization for Decentralized Networked SystemsabstractReinforcement learning algorithms require a large amount of samples; this often limits their real-world applications on even simple tasks. Such a challenge is more outstanding in multi-agent tasks, as each step of operation is more costly, requiring communications or shifting or resources. This work aims to improve data efficiency of multi-agent control by model-based learning. We consider networked systems where agents are cooperative and communicate only locally with their neighbors, and propose the decentralized model-based policy optimization framework (DMPO). In our method, each agent learns a dynamic model to predict future states and broadcast their predictions by communication, and then the policies are trained under the model rollouts. To alleviate the bias of model-generated data, we restrain the model usage for generating myopic rollouts, thus reducing the compounding error of model generation. To pertain the independence of policy update, we introduce extended value function and theoretically prove that the resulting policy gradient is a close approximation to true policy gradients. We evaluate our algorithm on several benchmarks for intelligent transportation systems, which are connected autonomous vehicle control tasks (Flow and CACC) and adaptive traffic signal control (ATSC). Empirical results show that our method achieves superior data efficiency and matches the performance of model-free methods using true models. The source code of our algorithm and baselines can be found at https://github.com/PKU-MARL/Model-Based-MARL. Yali Du 0001, Chengdong Ma, Runji Lin, Hao Dong 0003, Jun Wang 0012, Yaodong Yang 0001 |
IROS | 6 |
| 2022 | A Theoretical Understanding of Gradient Bias in Meta-Reinforcement LearningabstractGradient-based Meta-RL (GMRL) refers to methods that maintain two-level optimisation procedures wherein the outer-loop meta-learner guides the inner-loop gradient-based reinforcement learner to achieve fast adaptations. In this paper, we develop a unified framework that describes variations of GMRL algorithms and points out that existing stochastic meta-gradient estimators adopted by GMRL are actually \textbf{biased}. Such meta-gradient bias comes from two sources: 1) the compositional bias incurred by the two-level problem structure, which has an upper bound of $\mathcal{O}\big(K\alpha^{K}\hat{\sigma}_{\text{In}}|\tau|^{-0.5}\big)$ \emph{w.r.t.} inner-loop update step $K$, learning rate $\alpha$, estimate variance $\hat{\sigma}^{2}_{\text{In}}$ and sample size $|\tau|$, and 2) the multi-step Hessian estimation bias $\hat{\Delta}_{H}$ due to the use of autodiff, which has a polynomial impact $\mathcal{O}\big((K-1)(\hat{\Delta}_{H})^{K-1}\big)$ on the meta-gradient bias. We study tabular MDPs empirically and offer quantitative evidence that testifies our theoretical findings on existing stochastic meta-gradient estimators. Furthermore, we conduct experiments on Iterated Prisoner's Dilemma and Atari games to show how other methods such as off-policy learning and low-bias estimator can help fix the gradient bias for GMRL algorithms in general. Bo Liu 0039, Xidong Feng, Luo Mai, Haifeng Zhang 0002, Jun Wang 0012, Yaodong Yang 0001 |
NeurIPS | 7 |
| 2022 | Multiagent Q-learning with Sub-Team CoordinationabstractIn many real-world cooperative multiagent reinforcement learning (MARL) tasks, teams of agents can rehearse together before deployment, but then communication constraints may force individual agents to execute independently when deployed. Centralized training and decentralized execution (CTDE) is increasingly popular in recent years, focusing mainly on this setting. In the value-based MARL branch, credit assignment mechanism is typically used to factorize the team reward into each individual’s reward — individual-global-max (IGM) is a condition on the factorization ensuring that agents’ action choices coincide with team’s optimal joint action. However, current architectures fail to consider local coordination within sub-teams that should be exploited for more effective factorization, leading to faster learning. We propose a novel value factorization framework, called multiagent Q-learning with sub-team coordination (QSCAN), to flexibly represent sub-team coordination while honoring the IGM condition. QSCAN encompasses the full spectrum of sub-team coordination according to sub-team size, ranging from the monotonic value function class to the entire IGM function class, with familiar methods such as QMIX and QPLEX located at the respective extremes of the spectrum. Experimental results show that QSCAN’s performance dominates state-of-the-art methods in matrix games, predator-prey tasks, the Switch challenge in MA-Gym. Additionally, QSCAN achieves comparable performances to those methods in a selection of StarCraft II micro-management tasks. Wenhan Huang, Kai Li 0022, Kun Shao, Tianze Zhou, Matthew E. Taylor, Jun Luo 0009, Dongge Wang 0001, Hangyu Mao, Jianye Hao, Jun Wang 0012, Xiaotie Deng |
NeurIPS | 10 |
| 2022 | Optimistic Tree Searches for Combinatorial Black-Box OptimizationabstractThe optimization of combinatorial black-box functions is pervasive in computer science and engineering. However, the combinatorial explosion of the search space and lack of natural ordering pose significant challenges for current techniques from a theoretical and practical perspective, and require new algorithmic ideas. In this paper, we propose to adapt the recent advances in tree searches and partitioning techniques to design and analyze novel black-box combinatorial solvers. A first contribution is the analysis of a first tree-search algorithm called Optimistic Lipschitz Tree Search (OLTS) which assumes the Lipschitz constant of the function to be known. Linear convergence rates are provided for this algorithm under specific conditions, improving upon the logarithmic rates of baselines. An adaptive version, called Optimistic Combinatorial Tree Search (OCTS), is then introduced for the more realistic setup where we do not have any information on the Lipschitz constant of the function. Similar theoretical guarantees are shown to hold for OCTS and a numerical assessment is provided to illustrate the potential of tree searches with respect to state-of-the-art methods over typical benchmarks. Cédric Malherbe, Antoine Grosnit, Rasul Tutunov, Haitham Bou-Ammar, Jun Wang 0012 |
NeurIPS | 5 |
| 2022 | M2N: Mesh Movement Networks for PDE SolversabstractNumerical Partial Differential Equation (PDE) solvers often require discretizing the physical domain by using a mesh. Mesh movement methods provide the capability to improve the accuracy of the numerical solution without introducing extra computational burden to the PDE solver, by increasing mesh resolution where the solution is not well-resolved, whilst reducing unnecessary resolution elsewhere. However, sophisticated mesh movement methods, such as the Monge-Ampère method, generally require the solution of auxiliary equations. These solutions can be extremely expensive to compute when the mesh needs to be adapted frequently. In this paper, we propose to the best of our knowledge the first learning-based end-to-end mesh movement framework for PDE solvers. Key requirements of learning-based mesh movement methods are: alleviating mesh tangling, boundary consistency, and generalization to mesh with different resolutions. To achieve these goals, we introduce the neural spline model and the graph attention network (GAT) into our models respectively. While the Neural-Spline based model provides more flexibility for large mesh deformation, the GAT based model can handle domains with more complicated shapes and is better at performing delicate local deformation. We validate our methods on stationary and time-dependent, linear and non-linear equations, as well as regularly and irregularly shaped domains. Compared to the traditional Monge-Ampère method, our approach can greatly accelerate the mesh adaptation process by three to four orders of magnitude, whilst achieving comparable numerical error reduction. Wenbin Song, Joseph G. Wallwork, Junpeng Gao, Zheng Tian 0002, Fanglei Sun, Matthew D. Piggott, Zuoqiang Shi, Jun Wang 0012 |
NeurIPS | 11 |
| 2022 | Enhancing Safe Exploration Using Safety State AugmentationabstractSafe exploration is a challenging and important problem in model-free reinforcement learning (RL). Often the safety cost is sparse and unknown, which unavoidably leads to constraint violations - a phenomenon ideally to be avoided in safety-critical applications. We tackle this problem by augmenting the state-space with a safety state, which is nonnegative if and only if the constraint is satisfied. The value of this state also serves as a distance toward constraint violation, while its initial value indicates the available safety budget. This idea allows us to derive policies for scheduling the safety budget during training. We call our approach Simmer (Safe policy IMproveMEnt for RL) to reflect the careful nature of these schedules. We apply this idea to two safe RL problems: RL with constraints imposed on an average cost, and RL with constraints imposed on a cost with probability one. Our experiments suggest that "simmering" a safe algorithm can improve safety during training for both settings. We further show that Simmer can stabilize training and improve the performance of safe RL with average constraints. Aivar Sootla, Alexander I. Cowen-Rivers, Jun Wang 0012, Haitham Bou-Ammar |
NeurIPS | 3 |
| 2022 | Multi-Agent Reinforcement Learning is a Sequence Modeling ProblemabstractLarge sequence models (SM) such as GPT series and BERT have displayed outstanding performance and generalization capabilities in natural language process, vision and recently reinforcement learning. A natural follow-up question is how to abstract multi-agent decision making also as an sequence modeling problem and benefit from the prosperous development of the SMs. In this paper, we introduce a novel architecture named Multi-Agent Transformer (MAT) that effectively casts cooperative multi-agent reinforcement learning (MARL) into SM problems wherein the objective is to map agents' observation sequences to agents' optimal action sequences. Our goal is to build the bridge between MARL and SMs so that the modeling power of modern sequence models can be unleashed for MARL. Central to our MAT is an encoder-decoder architecture which leverages the multi-agent advantage decomposition theorem to transform the joint policy search problem into a sequential decision making process; this renders only linear time complexity for multi-agent problems and, most importantly, endows MAT with monotonic performance improvement guarantee. Unlike prior arts such as Decision Transformer fit only pre-collected offline data, MAT is trained by online trial and error from the environment in an on-policy fashion. To validate MAT, we conduct extensive experiments on StarCraftII, Multi-Agent MuJoCo, Dexterous Hands Manipulation, and Google Research Football benchmarks. Results demonstrate that MAT achieves superior performance and data efficiency compared to strong baselines including MAPPO and HAPPO. Furthermore, we demonstrate that MAT is an excellent few-short learner on unseen tasks regardless of changes in the number of agents.See our project page at https://sites.google.com/view/multi-agent-transformer. Muning Wen, Jakub Grudzien Kuba, Runji Lin, Weinan Zhang 0001, Ying Wen 0001, Jun Wang 0012, Yaodong Yang 0001 |
NeurIPS | 6 |
| 2022 | Branch Ranking for Efficient Mixed-Integer Programming via Offline Ranking-Based Policy Learning
Zeren Huang, Weinan Zhang 0001, Chuhan Shi, Furui Liu, Hui-Ling Zhen, Mingxuan Yuan, Jianye Hao, Yong Yu 0001, Jun Wang 0012 |
ECML/PKDD (5) | 10 |
| 2022 | Masked Gradient-Based Causal Structure LearningabstractThis paper studies the problem of learning causal structures from observational data. We reformulate the Structural Equation Model (SEM) with additive noises in a form parameterized by binary graph adjacency matrix and show that, if the original SEM is identifiable, then the binary adjacency matrix can be identified up to super-graphs of the true causal graph under mild conditions. We then utilize the reformulated SEM to develop a causal structure learning method that can be efficiently trained using gradient-based optimization, by leveraging a smooth characterization on acyclicity and the Gumbel-Softmax approach to approximate the binary adjacency matrix. It is found that the obtained entries are typically near zero or one and can be easily thresholded to identify the edges. We conduct experiments on synthetic and real datasets to validate the effectiveness of the proposed method, and show that it readily includes different smooth model functions and achieves a much improved performance on most datasets considered. Ignavier Ng, Shengyu Zhu 0001, Zhuangyan Fang, Haoyang Li 0002, Zhitang Chen, Jun Wang 0012 |
SDM | 6 |
| 2022 | Inhomogeneous deep Q-network for time sensitive applications
Xu Chen 0017, Jun Wang 0012 |
Artif. Intell. | 2 |
| 2022 | A bias-variance evaluation framework for information retrieval systems
Peng Zhang 0002, Zeting Hu, Dawei Song 0001, Jun Wang 0012, Yuexian Hou, Bin Hu 0001 |
Inf. Process. Manag. | 6 |
| 2022 | HEBO: An Empirical Study of Assumptions in Bayesian OptimisationabstractIn this work we rigorously analyse assumptions inherent to black-box optimisation hyper-parameter tuning tasks. Our results on the Bayesmark benchmark indicate that heteroscedasticity and non-stationarity pose significant challenges for black-box optimisers. Based on these findings, we propose a Heteroscedastic and Evolutionary Bayesian Optimisation solver (HEBO). HEBO performs non-linear input and output warping, admits exact marginal log-likelihood optimisation and is robust to the values of learned parameters. We demonstrate HEBO’s empirical efficacy on the NeurIPS 2020 Black-Box Optimisation challenge, where HEBO placed first. Upon further analysis, we observe that HEBO significantly outperforms existing black-box optimisers on 108 machine learning hyperparameter tuning tasks comprising the Bayesmark benchmark. Our findings indicate that the majority of hyper-parameter tuning tasks exhibit heteroscedasticity and non-stationarity, multiobjective acquisition ensembles with Pareto front solutions improve queried configurations, and robust acquisition maximisers afford empirical advantages relative to their non-robust counterparts. We hope these findings may serve as guiding principles for practitioners of Bayesian optimisation. Alexander I. Cowen-Rivers, Wenlong Lyu, Rasul Tutunov, Zhi Wang 0001, Antoine Grosnit, Ryan-Rhys Griffiths, Alexandre Maraval, Jianye Hao, Jun Wang 0012, Jan Peters 0001, Haitham Bou-Ammar |
J. Artif. Intell. Res. | 9 |
| 2022 | SAMBA: safe model-based & active reinforcement learning
Alexander I. Cowen-Rivers, Daniel Palenicek, Vincent Moens, Mohammed Amin Abdullah 0001, Aivar Sootla, Jun Wang 0012, Haitham Bou-Ammar |
Mach. Learn. | 6 |
| 2022 | Learning to select cuts for efficient mixed-integer programming
Zeren Huang, Kerong Wang, Furui Liu, Hui-Ling Zhen, Weinan Zhang 0001, Mingxuan Yuan, Jianye Hao, Yong Yu 0001, Jun Wang 0012 |
Pattern Recognit. | 9 |
| 2022 | Enhanced Discrete Multi-Modal Hashing: More Constraints Yet Less Time to LearnabstractDue to the exponential growth of multimedia data, multi-modal hashing as a promising technique to make cross-view retrieval scalable is attracting more and more attention. However, most of the existing multi-modal hashing methods either divide the learning process unnaturally into two separate stages or treat the discrete optimization problem simplistically as a continuous one, which leads to suboptimal results. Recently, a few discrete multi-modal hashing methods that try to address such issues have emerged, but they still ignore several important discrete constraints (such as the balance and decorrelation of hash bits). In this paper, we overcome those limitations by proposing a novel method named “Enhanced Discrete Multi-modal Hashing (EDMH)” which learns binary codes and hashing functions simultaneously from the pairwise similarity matrix of data, under the aforementioned discrete constraints. Although the model of EDMH looks a lot more complex than the other models for multi-modal hashing, we are actually able to develop a fast iterative learning algorithm for it, since the subproblems of its optimization all have closed-form solutions after introducing a couple of auxiliary variables. Our experimental results on three real-world datasets have revealed the usefulness of those previously ignored discrete constraints and demonstrated that EDMH not only performs much better than state-of-the-art competitors according to several retrieval metrics but also runs much faster than most of them. Yong Chen 0008, Hui Zhang 0028, Zhibao Tian, Jun Wang 0012, Dell Zhang, Xuelong Li 0001 |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2022 | AutoHash: Learning Higher-Order Feature Interactions for Deep CTR PredictionabstractFeature combinations are essential for the success of many web applications, such as personalised recommendation and online advertising. State-of-the-art methods usually model explicit feature interactions to help neural networks reduce the number of parameters and achieve better performance. However, their explicit feature interactions are often restricted to the second-order due to computational complexity. In this work, we propose efficient ways to represent explicit high-order feature combinations as well as prune redundant features in the mean time. To begin with, we make novel use of the Count Sketch algorithm within a DNN classifier such that high-order feature combinations can be compactly represented. After that, to combat the problem of redundant features which degrade the prediction performance, we introduce an adaptive hashing algorithm, AutoHash, which can automatically select meaningful features to interact at high orders according to the specific dataset in question. This is an AutoML approach. Experiments on three well-known public datasets demonstrate that AutoHash is significantly superior to state-of-the-art methods. Meanwhile, due to its efficient scheme of automatically selecting useful high-order feature interactions, AutoHash has less model complexity and can be trained in an end-to-end manner with less training time than state-of-the-art methods. Niannan Xue, Bin Liu 0072, Huifeng Guo, Ruiming Tang, Fengwei Zhou, Stefanos Zafeiriou, Jun Wang 0012, Zhenguo Li |
IEEE Trans. Knowl. Data Eng. | 8 |
| 2022 | Beyond Relevance Ranking: A General Graph Matching Framework for Utility-Oriented Learning to RankabstractLearning to rank from logged user feedback, such as clicks or purchases, is a central component of many real-world information systems. Different from human-annotated relevance labels, the user feedback is always noisy and biased. Many existing learning to rank methods infer the underlying relevance of query–item pairs based on different assumptions of examination, and still optimize a relevance based objective. Such methods rely heavily on the correct estimation of examination, which is often difficult to achieve in practice. In this work, we propose a general framework U-rank+ for learning to rank with logged user feedback from the perspective of graph matching. We systematically analyze the biases in user feedback, including examination bias and selection bias. Then, we take both biases into consideration for unbiased utility estimation that directly based on user feedback, instead of relevance. In order to maximize the estimated utility in an efficient manner, we design two different solvers based on Sinkhorn and LambdaLoss for U-rank+ . The former is based on a standard graph matching algorithm, and the latter is inspired by the traditional method of learning to rank. Both of the algorithms have good theoretical properties to optimize the unbiased utility objective while the latter is proved to be empirically more effective and efficient in practice. Our framework U-rank+ can deal with a general utility function and can be used in a widespread of applications including web search, recommendation, and online advertising. Semi-synthetic experiments on three benchmark learning to rank datasets demonstrate the effectiveness of U-rank+ . Furthermore, our proposed framework has been deployed on two different scenarios of a mainstream App store, where the online A/B testing shows that U-rank+ achieves an average improvement of 19.2% on click-through rate and 20.8% improvement on conversion rate in recommendation scenario, and 5.12% on platform revenue in online advertising scenario over the production baselines. Xinyi Dai, Yunjia Xi, Weinan Zhang 0001, Qing Liu 0020, Ruiming Tang, Xiuqiang He 0001, Jun Wang 0012, Yong Yu 0001 |
ACM Trans. Inf. Syst. | 8 |
| 2022 | Multi-Agent Feedback Enabled Neural Networks for Intelligent CommunicationsabstractIn the intelligent communication field, deep learning (DL) has attracted much attention due to its strong fitting ability and data-driven learning capability. Compared with the typical DL feedforward network structures, an enhancement structure with direct data feedback have been studied and proved to have better performance than the feedfoward networks. However, due to the above simple feedback methods lack sufficient analysis and learning ability on the feedback data, it is inadequate to deal with more complicated nonlinear systems and therefore the performance is limited for further improvement. In this paper, a novel multi-agent feedback enabled neural network (MAFENN) framework is proposed, consisting of three fully cooperative intelligent agents, which make the framework have stronger feedback learning capabilities and more intelligence on feature abstraction, denoising or generation, etc. Furthermore, the MAFENN frame work is theoretically formulated into a three-player Feedback Stackelberg game, and the game is proved to converge to the Feedback Stackelberg equilibrium. The design of MAFENN framework and algorithm are dedicated to enhance the learning capability of the feedfoward DL networks or their variations with the simple data feedback. To verify the MAFENN framework’s feasibility in wireless communications, a multi-agent MAFENN based equalizer (MAFENN-E) is developed for wireless fading channels with inter-symbol interference (ISI). Experimental results show that when the quadrature phase-shift keying (QPSK) modulation scheme is adopted, the SER performance of our proposed method outperforms that of the traditional equalizers by about 2 dB in linear channels. When in nonlinear channels, the SER performance of our proposed method outperforms that of either traditional or DL based equalizers more significantly, which shows the effectiveness and robustness of our proposal in the complex channel environment. Fanglei Sun, Yang Li 0116, Ying Wen 0001, Jingchen Hu, Jun Wang 0012, Yang Yang 0001, Kai Li 0022 |
IEEE Trans. Wirel. Commun. | 5 |
| 2021 | CMML: Contextual Modulation Meta Learning for Cold-Start RecommendationabstractPractical recommender systems experience a cold-start problem when observed user-item interactions in the history are insufficient. Meta learning, especially gradient based one, can be adopted to tackle this problem by learning initial parameters of the model and thus allowing fast adaptation to a specific task from limited data examples. Though with significant performance improvement, it commonly suffers from two critical issues: the non-compatibility with mainstream industrial deployment and the heavy computational burdens, both due to the inner-loop gradient operation. These two issues make them hard to be applied in practical recommender systems. To enjoy the benefits of meta learning framework and mitigate these problems, we propose a recommendation framework called Contextual Modulation Meta Learning (CMML). CMML is composed of fully feed-forward operations so it is computationally efficient and completely compatible with the mainstream industrial deployment. CMML consists of three components, including a context encoder that can generate context embedding to represent a specific task, a hybrid context generator that aggregates specific user-item features with task-level context, and a contextual modulation network, which can modulate the recommendation model to adapt effectively. We validate our approach on both scenario-specific and user-specific cold-start setting on various real-world datasets, showing CMML can achieve comparable or even better performance with gradient based methods yet with higher computational efficiency and better interpretability. Xidong Feng, Chen Chen 0077, Dong Li 0016, Mengchen Zhao, Jianye Hao, Jun Wang 0012 |
CIKM | 6 |
| 2021 | Grassland: A Rapid Algebraic Modeling System for Million-variable OptimizationabstractAn algebraic modeling system (AMS) is a type of mathematical software for optimization problems, which allows users to define symbolic mathematical models in a specific language, instantiate them with given source of data, and solve them with the aid of external solver engines. With the bursting scale of business models and increasing need for timeliness, traditional AMSs are not sufficient to meet the following industry needs: 1) million-variable models need to be instantiated from raw data very efficiently; 2) Strictly feasible solution of million-variable models need to be delivered in a rapid manner to make up-to-date decisions against highly dynamic environments. Grassland is a rapid AMS that provides an end-to-end solution to tackle these emerged new challenges. It integrates a parallelized instantiation scheme for large-scale linear constraints, and a sequential decomposition method that accelerates model solving exponentially with an acceptable loss of optimality. Extensive benchmarks on both classical models and real enterprise scenario demonstrate 6-10x speedup of Grassland over state-of-the-art solutions on model instantiation. Our proposed system has been deployed in the large-scale real production planning scenario of Huawei. With the aid of our decomposition method, Grassland successfully accelerated Huawei's million-variable production planning simulation pipeline from hours to 3-5 minutes, supporting near-real-time production plan decision making against highly dynamic supply-demand environment. Xihan Li 0001, Xiongwei Han, Zhishuo Zhou, Mingxuan Yuan, Jun Wang 0012 |
CIKM | 6 |
| 2021 | Top-N Recommendation with Counterfactual User Preference SimulationabstractTop-N recommendation, which aims to learn user ranking-based preference, has long been a fundamental problem in a wide range of applications. Traditional models usually motivate themselves by designing complex or tailored architectures based on different assumptions. However, the training data of recommender system can be extremely sparse and imbalanced, which poses great challenges for boosting the recommendation performance. To alleviate this problem, in this paper, we propose to reformulate the recommendation task within the causal inference framework, which enables us to counterfactually simulate user ranking-based preferences to handle the data scarce problem. The core of our model lies in the counterfactual question: "what would be the user's decision if the recommended items had been different?''. To answer this question, we firstly formulate the recommendation process with a series of structural equation models (SEMs), whose parameters are optimized based on the observed data. Then, we actively indicate many recommendation lists (called intervention in the causal inference terminology) which are not recorded in the dataset, and simulate user feedback according to the learned SEMs for generating new training samples. Instead of randomly intervening on the recommendation list, we design a learning-based method to discover more informative training samples. Considering that the learned SEMs can be not perfect, we, at last, theoretically analyze the relation between the number of generated samples and the model prediction error, based on which a heuristic method is designed to control the negative effect brought by the prediction error. Extensive experiments are conducted based on both synthetic and real-world datasets to demonstrate the effectiveness of our framework. Mengyue Yang, Quanyu Dai, Zhenhua Dong, Xu Chen 0017, Xiuqiang He 0001, Jun Wang 0012 |
CIKM | 6 |
| 2021 | CausalVAE: Disentangled Representation Learning via Neural Structural Causal ModelsabstractLearning disentanglement aims at finding a low dimensional representation which consists of multiple explanatory and generative factors of the observational data. The framework of variational autoencoder (VAE) is commonly used to disentangle independent factors from observations. However, in real scenarios, factors with semantics are not necessarily independent. Instead, there might be an underlying causal structure which renders these factors dependent. We thus propose a new VAE based framework named CausalVAE, which includes a Causal Layer to transform independent exogenous factors into causal endogenous ones that correspond to causally related concepts in data. We further analyze the model identifiabitily, showing that the proposed model learned from observations recovers the true one up to a certain degree. Experiments are conducted on various datasets, including synthetic and real word benchmark CelebA. Results show that the causal representations learned by CausalVAE are semantically interpretable, and their causal relationship as a Directed Acyclic Graph (DAG) is identified with good accuracy. Furthermore, we demonstrate that the proposed CausalVAE model is able to generate counterfactual data through "do-operation" to the causal factors. Mengyue Yang, Furui Liu, Zhitang Chen, Xinwei Shen 0002, Jianye Hao, Jun Wang 0012 |
CVPR | 6 |
| 2021 | MAFENN: Multi-Agent Feedback Enabled Neural Network for Wireless Channel EqualizationabstractFeedback mechanism has been widely used in wireless communication such as channel equalization and resource allocation. In recent years, deep learning (DL) has made great progress in the field of wireless communication. There is now some work that attempts to introduce plain feedback mechanisms into DL algorithm to solve wireless communication problems. However, the improvement of plain feedback DL methods is limited in complex situations due to those methods lack sufficient learning ability on feedback information. In this paper, we propose a Multi-Agent Feedback Enabled Neural Network (MAFENN) equalizer, which consists of a specific learnable feedback agent and two feed-forward agents. Three fully cooperative intelligent agents help the system improve the ability to remove wireless inter-symbol interference (ISI) in receiving ends. We further formulate it into a three-player Stackelberg Game, which helps us to optimize and train this model more efficiently. To verify the feasibility of our proposed MAFENN system and the Stackelberg Game optimization, we conduct a series of experiments to compare the symbol error rate (SER) performance of the MAFENN equalizer and the other methods which utilizes quadrature phase-shift keying (QPSK) modulation scheme. Our performance outperforms that of the other equalizers at different signal-to-noise ratio (SNR) settings for both linear and nonlinear channels. Yang Li 0116, Fanglei Sun, Weiqin Zu, Wenbin Song, Ying Wen 0001, Jun Wang 0012, Yang Yang 0001, Kai Li 0022, Liantao Wu |
GLOBECOM | 6 |
| 2021 | Joint Caching and Transmission in the Mobile Edge Network: An Multi-Agent Learning ApproachabstractJoint caching and transmission optimization problem is challenging due to the deep coupling between decisions. This paper proposes an iterative distributed multi-agent learning approach to jointly optimize caching and transmission. The goal of this approach is to minimize the total transmission delay of all users. In this iterative approach, each iteration includes caching optimization and transmission optimization. A multi-agent reinforcement learning (MARL)-based caching network is developed to cache popular tasks, such as answering which files to evict from the cache and which files to storage. Based on the cached files of the caching network, the transmission network transmits cached files for users by single transmission (ST) or joint transmission (JT) with multi-agent Bayesian learning automaton (MABLA) method. And then users access the edge servers with the minimum transmission delay. The experimental results demonstrate the performance of the proposed multi-agent learning approach. Qirui Mi, Ning Yang 0005, Haifeng Zhang 0002, Haijun Zhang 0001, Jun Wang 0012 |
GLOBECOM | 5 |
| 2021 | Retrospective Thinking based Multi-Agent System for Wireless Video TransmissionsabstractBenefiting from the breakthrough development of the fifth generation (5G), beyond 5G (B5G) wireless communication networks and Artificial Intelligence (AI) in recent years, the artificial intelligence of things (AIoT) is a new trend in the future. AIoT devices often have high-quality wireless video transmission requirements. However, the propagating signals at millimeter wave suffer from high propagation loss and sensitivity to blockage, resulting in the received video is vulnerable to be interfered. Due to the ability of Deep Learning (DL) to discover and learn good representations, some DL methods have achieved breakthrough performance in video recovery. However, most of these methods cannot exploit information from the higher to lower level to refine themselves. In this paper, we propose a novel retrospective thinking based multi-agent (ReTMA) system to solve the interference problem experienced on wireless channels. Compared with other plain feedback models, we add a retrospective agent on the feedback loop, which makes the entire system have stronger capabilities to learn good representative features. We further formulate it as a Stackelberg game to analyze the dependency relationship between the agents and facilitate the complex training issue of the multiple agents. To verify the feasibility of ReTMA system, we randomly add masks to simulate the severe interference received by the video frames in wireless transmissions. Experimental results show that the performances of similarity index measure (SSIM), peak signal-to-noise ratio (PSNR) and classification accuracy all achieve significant gains compared with those of other plain feedback models at different mask ratios. Yang Li 0116, Fanglei Sun, Wenbin Song, Ying Wen 0001, Kai Li 0022, Jun Wang 0012, Yang Yang 0001 |
ICC | 6 |
| 2021 | Learning to Optimize Industry-Scale Dynamic Pickup and Delivery ProblemsabstractThe Dynamic Pickup and Delivery Problem (DPDP) is aimed at dynamically scheduling vehicles among multiple sites in order to minimize the cost when delivery orders are not known a priori. Although DPDP plays an important role in modern logistics and supply chain management, state-of-the-art DPDP algorithms are still limited on their solution quality and efficiency. In practice, they fail to provide a scalable solution as the numbers of vehicles and sites become large. In this paper, we propose a data-driven approach, Spatial-Temporal Aided Double Deep Graph Network (ST-DDGN), to solve industry-scale DPDP. In our method, the delivery demands are first forecast using spatial-temporal prediction method, which guides the neural network to perceive spatial-temporal distribution of delivery demand when dispatching vehicles. Besides, the relationships of individuals such as vehicles are modelled by establishing a graph-based value function. ST-DDGN incorporates attention-based graph embedding with Double DQN (DDQN). As such, it can make the inference across vehicles more efficiently compared with traditional methods. Our method is entirely data driven and thus adaptive, i.e., the relational representation of adjacent vehicles can be learned and corrected by ST-DDGN from data periodically. We have conducted extensive experiments over real-world data to evaluate our solution. The results show that ST-DDGN reduces 11.27% number of the used vehicles and decreases 13.12% total transportation cost on average over the strong baselines, including the heuristic algorithm deployed in our UAT (User Acceptance Test) environment and a variety of vanilla DRL methods. We are due to fully deploy our solution into our online logistics system and it is estimated that millions of USD logistics cost can be saved per year. Xijun Li, Weilin Luo, Mingxuan Yuan, Jun Wang 0012, Jie Wang 0005, Jinhu Lü 0001 |
ICDE | 4 |
| 2021 | Estimating α-Rank from A Few Entries with Low Rank Matrix CompletionabstractMulti-agent evaluation aims at the assessment of an agent’s strategy on the basis of interaction with others. Typically, existing methods such as $\alpha$-rank and its approximation still require to exhaustively compare all pairs of joint strategies for an accurate ranking, which in practice is computationally expensive. In this paper, we aim to reduce the number of pairwise comparisons in recovering a satisfying ranking for $n$ strategies in two-player meta-games, by exploring the fact that agents with similar skills may achieve similar payoffs against others. Two situations are considered: the first one is when we can obtain the true payoffs; the other one is when we can only access noisy payoff. Based on these formulations, we leverage low-rank matrix completion and design two novel algorithms for noise-free and noisy evaluations respectively. For both of these settings, we theorize that $O(nr \log n)$ ($n$ is the number of agents and $r$ is the rank of the payoff matrix) payoff entries are required to achieve sufficiently well strategy evaluation performance. Empirical results on evaluating the strategies in three synthetic games and twelve real world games demonstrate that strategy evaluation from a few entries can lead to comparable performance to algorithms with full knowledge of the payoff matrix. Yali Du 0001, Xu Chen 0017, Jun Wang 0012, Haifeng Zhang 0002 |
ICML | 4 |
| 2021 | Learning in Nonzero-Sum Stochastic Games with PotentialsabstractMulti-agent reinforcement learning (MARL) has become effective in tackling discrete cooperative game scenarios. However, MARL has yet to penetrate settings beyond those modelled by team and zero-sum games, confining it to a small subset of multi-agent systems. In this paper, we introduce a new generation of MARL learners that can handle \textit{nonzero-sum} payoff structures and continuous settings. In particular, we study the MARL problem in a class of games known as stochastic potential games (SPGs) with continuous state-action spaces. Unlike cooperative games, in which all agents share a common reward, SPGs are capable of modelling real-world scenarios where agents seek to fulfil their individual goals. We prove theoretically our learning method, $\ourmethod$, enables independent agents to learn Nash equilibrium strategies in \textit{polynomial time}. We demonstrate our framework tackles previously unsolvable tasks such as \textit{Coordination Navigation} and \textit{large selfish routing games} and that it outperforms the state of the art MARL baselines such as MADDPG and COMIX in such scenarios. David Mguni, Yutong Wu 0005, Yali Du 0001, Yaodong Yang 0001, Ziyi Wang 0004, Minne Li, Ying Wen 0001, Joel Jennings, Jun Wang 0012 |
ICML | 9 |
| 2021 | Modelling Behavioural Diversity for Learning in Open-Ended GamesabstractPromoting behavioural diversity is critical for solving games with non-transitive dynamics where strategic cycles exist, and there is no consistent winner (e.g., Rock-Paper-Scissors). Yet, there is a lack of rigorous treatment for defining diversity and constructing diversity-aware learning dynamics. In this work, we offer a geometric interpretation of behavioural diversity in games and introduce a novel diversity metric based on \emph{determinantal point processes} (DPP). By incorporating the diversity metric into best-response dynamics, we develop \emph{diverse fictitious play} and \emph{diverse policy-space response oracle} for solving normal-form games and open-ended games. We prove the uniqueness of the diverse best response and the convergence of our algorithms on two-player games. Importantly, we show that maximising the DPP-based diversity metric guarantees to enlarge the \emph{gamescape} – convex polytopes spanned by agents’ mixtures of strategies. To validate our diversity-aware solvers, we test on tens of games that show strong non-transitivity. Results suggest that our methods achieve at least the same, and in most games, lower exploitability than PSRO solvers by finding effective and diverse strategies. Nicolas Perez Nieves, Yaodong Yang 0001, Oliver Slumbers, David Mguni, Ying Wen 0001, Jun Wang 0012 |
ICML | 6 |
| 2021 | Ordering-Based Causal Discovery with Reinforcement LearningabstractIt is a long-standing question to discover causal relations among a set of variables in many empirical sciences. Recently, Reinforcement Learning (RL) has achieved promising results in causal discovery from observational data. However, searching the space of directed graphs and enforcing acyclicity by implicit penalties tend to be inefficient and restrict the existing RL-based method to small scale problems. In this work, we propose a novel RL-based approach for causal discovery, by incorporating RL into the ordering-based paradigm. Specifically, we formulate the ordering search problem as a multi-step Markov decision process, implement the ordering generating process with an encoder-decoder architecture, and finally use RL to optimize the proposed model based on the reward mechanisms designed for each ordering. A generated ordering would then be processed using variable selection to obtain the final causal graph. We analyze the consistency and computational complexity of the proposed method, and empirically show that a pretrained model can be exploited to accelerate training. Experimental results on both synthetic and real data sets shows that the proposed method achieves a much improved performance over existing RL-based method. Xiaoqiang Wang 0003, Yali Du 0001, Shengyu Zhu 0001, Liangjun Ke, Zhitang Chen, Jianye Hao, Jun Wang 0012 |
IJCAI | 7 |
| 2021 | State-Aware Value Function Approximation with Attention Mechanism for Restless Multi-armed BanditsabstractThe restless multi-armed bandit (RMAB) problem is a generalization of the multi-armed bandit with non-stationary rewards. Its optimal solution is intractable due to exponentially large state and action spaces with respect to the number of arms. Existing approximation approaches, e.g., Whittle's index policy, have difficulty in capturing either temporal or spatial factors such as impacts from other arms. We propose considering both factors using the attention mechanism, which has achieved great success in deep learning. Our state-aware value function approximation solution comprises an attention-based value function approximator and a Bellman equation solver. The attention-based coordination module capture both spatial and temporal factors for arm coordination. The Bellman equation solver utilizes the decoupling structure of RMABs to acquire solutions with significantly reduced computation overheads. In particular, the time complexity of our approximation is linear in the number of arms. Finally, we illustrate the effectiveness and investigate the properties of our proposed method with numerical experiments. Shuang Wu 0005, Jingyu Zhao 0001, Guangjian Tian, Jun Wang 0012 |
IJCAI | 4 |
| 2021 | Neural Auto-Curricula in Two-Player Zero-Sum GamesabstractWhen solving two-player zero-sum games, multi-agent reinforcement learning (MARL) algorithms often create populations of agents where, at each iteration, a new agent is discovered as the best response to a mixture over the opponent population. Within such a process, the update rules of "who to compete with" (i.e., the opponent mixture) and "how to beat them" (i.e., finding best responses) are underpinned by manually developed game theoretical principles such as fictitious play and Double Oracle. In this paper, we introduce a novel framework—Neural Auto-Curricula (NAC)—that leverages meta-gradient descent to automate the discovery of the learning update rule without explicit human design. Specifically, we parameterise the opponent selection module by neural networks and the best-response module by optimisation subroutines, and update their parameters solely via interaction with the game engine, where both players aim to minimise their exploitability. Surprisingly, even without human design, the discovered MARL algorithms achieve competitive or even better performance with the state-of-the-art population-based game solvers (e.g., PSRO) on Games of Skill, differentiable Lotto, non-transitive Mixture Games, Iterated Matching Pennies, and Kuhn Poker. Additionally, we show that NAC is able to generalise from small games to large games, for example training on Kuhn Poker and outperforming PSRO on Leduc Poker. Our work inspires a promising future direction to discover general MARL algorithms solely from data. Xidong Feng, Oliver Slumbers, Ziyu Wan, Bo Liu 0039, Stephen McAleer, Ying Wen 0001, Jun Wang 0012, Yaodong Yang 0001 |
NeurIPS | 7 |
| 2021 | Settling the Variance of Multi-Agent Policy GradientsabstractPolicy gradient (PG) methods are popular reinforcement learning (RL) methods where a baseline is often applied to reduce the variance of gradient estimates. In multi-agent RL (MARL), although the PG theorem can be naturally extended, the effectiveness of multi-agent PG (MAPG) methods degrades as the variance of gradient estimates increases rapidly with the number of agents. In this paper, we offer a rigorous analysis of MAPG methods by, firstly, quantifying the contributions of the number of agents and agents' explorations to the variance of MAPG estimators. Based on this analysis, we derive the optimal baseline (OB) that achieves the minimal variance. In comparison to the OB, we measure the excess variance of existing MARL algorithms such as vanilla MAPG and COMA. Considering using deep neural networks, we also propose a surrogate version of OB, which can be seamlessly plugged into any existing PG methods in MARL. On benchmarks of Multi-Agent MuJoCo and StarCraft challenges, our OB technique effectively stabilises training and improves the performance of multi-agent PPO and COMA algorithms by a significant margin. Code is released at \url{https://github.com/morning9393/Optimal-Baseline-for-Multi-agent-Policy-Gradients}. Jakub Grudzien Kuba, Muning Wen, Linghui Meng 0001, Shangding Gu, Haifeng Zhang 0002, David Mguni, Jun Wang 0012, Yaodong Yang 0001 |
NeurIPS | 7 |
| 2021 | Reinforcement Recommendation with User Multi-aspect PreferenceabstractFormulating recommender system with reinforcement learning (RL) frameworks has attracted increasing attention from both academic and industry communities. While many promising results have been achieved, existing models mostly simulate the environment reward with a unified value, which may hinder the understanding of users’ complex preferences and limit the model performance. In this paper, we consider how to model user multi-aspect preferences in the context of RL-based recommender system. More specifically, we base our model on the framework of deterministic policy gradient (DPG), which is effective in dealing with large action spaces. A major challenge for modeling user multi-aspect preferences lies in the fact that they may contradict with each other. To solve this problem, we introduce Pareto optimization into the DPG framework. We assign each aspect with a tailored critic, and all the critics share the same actor. The Pareto optimization is realized by a gradient-based method, which can be easily integrated into the actor and critic learning process. Based on the designed model, we theoretically analyze its gradient bias in the optimization process, and we design a weight-reuse mechanism to lower the upper bound of this bias, which is shown to be effective for improving the model performance. We conduct extensive experiments based on three real-world datasets to demonstrate our model’s superiorities. Xu Chen 0017, Yali Du 0001, Jun Wang 0012 |
WWW | 4 |
| 2021 | An Adversarial Imitation Click Model for Information RetrievalabstractModern information retrieval systems, including web search, ads placement, and recommender systems, typically rely on learning from user feedback. Click models, which study how users interact with a ranked list of items, provide a useful understanding of user feedback for learning ranking models. Constructing ”right” dependencies is the key of any successful click model. However, probabilistic graphical models (PGMs) have to rely on manually assigned dependencies, and oversimplify user behaviors. Existing neural network based methods promote PGMs by enhancing the expressive ability and allowing flexible dependencies, but still suffer from exposure bias and inferior estimation. In this paper, we propose a novel framework, Adversarial Imitation Click Model (AICM), based on imitation learning. Firstly, we explicitly learn the reward function that recovers users’ intrinsic utility and underlying intentions. Secondly, we model user interactions with a ranked list as a dynamic system instead of one-step click prediction, alleviating the exposure bias problem. Finally, we minimize the JS divergence through adversarial training and learn a stable distribution of click sequences, which makes AICM generalize well across different distributions of ranked lists. A theoretical analysis has indicated that AICM reduces the exposure bias from O(T2) to O(T). Our studies on a public web search dataset show that AICM not only outperforms state-of-the-art models in traditional click metrics but also achieves superior performance in addressing the exposure bias and recovering the underlying patterns of click sequences. Xinyi Dai, Jianghao Lin, Weinan Zhang 0001, Shuai Li 0010, Weiwen Liu, Ruiming Tang, Xiuqiang He 0001, Jianye Hao, Jun Wang 0012, Yong Yu 0001 |
WWW | 9 |
| 2021 | Are We Forgetting about Compositional Optimisers in Bayesian Optimisation?abstractBayesian optimisation presents a sample-efficient methodology for global optimisation. Within this framework, a crucial performance-determining subroutine is the maximisation of the acquisition function, a task complicated by the fact that acquisition functions tend to be non-convex and thus nontrivial to optimise. In this paper, we undertake a comprehensive empirical study of approaches to maximise the acquisition function. Additionally, by deriving novel, yet mathematically equivalent, compositional forms for popular acquisition functions, we recast the maximisation task as a compositional optimisation problem, allowing us to benefit from the extensive literature in this field. We highlight the empirical advantages of the compositional approach to acquisition function maximisation across 3958 individual experiments comprising synthetic optimisation tasks as well as tasks from Bayesmark. Given the generality of the acquisition function maximisation subroutine, we posit that the adoption of compositional optimisers has the potential to yield performance improvements across all domains in which Bayesian optimisation is currently being applied. An open-source implementation is made available at https://github.com/huawei-noah/noah-research/tree/CompBO/BO/HEBO/CompBO. Antoine Grosnit, Alexander I. Cowen-Rivers, Rasul Tutunov, Ryan-Rhys Griffiths, Jun Wang 0012, Haitham Bou-Ammar |
J. Mach. Learn. Res. | 5 |
| 2020 | Learning to Model Opponent Learning (Student Abstract)abstractMulti-Agent Reinforcement Learning (MARL) considers settings in which a set of coexisting agents interact with one another and their environment. The adaptation and learning of other agents induces non-stationarity in the environment dynamics. This poses a great challenge for value function-based algorithms whose convergence usually relies on the assumption of a stationary environment. Policy search algorithms also struggle in multi-agent settings as the partial observability resulting from an opponent's actions not being known introduces high variance to policy training. Modelling an agent's opponent(s) is often pursued as a means of resolving the issues arising from the coexistence of learning opponents. An opponent model provides an agent with some ability to reason about other agents to aid its own decision making. Most prior works learn an opponent model by assuming the opponent is employing a stationary policy or switching between a set of stationary policies. Such an approach can reduce the variance of training signals for policy search algorithms. However, in the multi-agent setting, agents have an incentive to continually adapt and learn. This means that the assumptions concerning opponent stationarity are unrealistic. In this work, we develop a novel approach to modelling an opponent's learning dynamics which we term Learning to Model Opponent Learning (LeMOL). We show our structured opponent model is more accurate and stable than naive behaviour cloning baselines. We further show that opponent modelling can improve the performance of algorithmic agents in multi-agent settings. Ian Davies, Zheng Tian 0002, Jun Wang 0012 |
AAAI | 3 |
| 2020 | Neighborhood Cognition Consistent Multi-Agent Reinforcement LearningabstractSocial psychology and real experiences show that cognitive consistency plays an important role to keep human society in order: if people have a more consistent cognition about their environments, they are more likely to achieve better cooperation. Meanwhile, only cognitive consistency within a neighborhood matters because humans only interact directly with their neighbors. Inspired by these observations, we take the first step to introduce neighborhood cognitive consistency (NCC) into multi-agent reinforcement learning (MARL). Our NCC design is quite general and can be easily combined with existing MARL methods. As examples, we propose neighborhood cognition consistent deep Q-learning and Actor-Critic to facilitate large-scale multi-agent cooperations. Extensive experiments on several challenging tasks (i.e., packet routing, wifi configuration and Google football player control) justify the superior performance of our methods compared with state-of-the-art MARL approaches. Hangyu Mao, Wulong Liu, Jianye Hao, Jun Luo 0009, Dong Li 0016, Zhengchao Zhang, Jun Wang 0012 |
AAAI | 7 |
| 2020 | Learning to Communicate Implicitly by ActionsabstractIn situations where explicit communication is limited, human collaborators act by learning to: (i) infer meaning behind their partner's actions, and (ii) convey private information about the state to their partner implicitly through actions. The first component of this learning process has been well-studied in multi-agent systems, whereas the second — which is equally crucial for successful collaboration — has not. To mimic both components mentioned above, thereby completing the learning process, we introduce a novel algorithm: Policy Belief Learning (PBL). PBL uses a belief module to model the other agent's private information and a policy module to form a distribution over actions informed by the belief module. Furthermore, to encourage communication by actions, we propose a novel auxiliary reward which incentivizes one agent to help its partner to make correct inferences about its private information. The auxiliary reward for communication is integrated into the learning of the policy module. We evaluate our approach on a set of environments including a matrix game, particle environment and the non-competitive bidding problem from contract bridge. We show empirically that this auxiliary reward is effective and easy to generalize. These results demonstrate that our PBL algorithm can produce strong pairs of agents in collaborative games where explicit communication is disabled. Zheng Tian 0002, Shihao Zou, Ian Davies, Tim Warr, Lisheng Wu, Haitham Bou-Ammar, Jun Wang 0012 |
AAAI | 7 |
| 2020 | Bi-Level Actor-Critic for Multi-Agent CoordinationabstractCoordination is one of the essential problems in multi-agent systems. Typically multi-agent reinforcement learning (MARL) methods treat agents equally and the goal is to solve the Markov game to an arbitrary Nash equilibrium (NE) when multiple equilibra exist, thus lacking a solution for NE selection. In this paper, we treat agents unequally and consider Stackelberg equilibrium as a potentially better convergence point than Nash equilibrium in terms of Pareto superiority, especially in cooperative environments. Under Markov games, we formally define the bi-level reinforcement learning problem in finding Stackelberg equilibrium. We propose a novel bi-level actor-critic learning method that allows agents to have different knowledge base (thus intelligent), while their actions still can be executed simultaneously and distributedly. The convergence proof is given, while the resulting learning algorithm is tested against the state of the arts. We found that the proposed bi-level actor-critic algorithm successfully converged to the Stackelberg equilibria in matrix games and find a asymmetric solution in a highway merge environment. Haifeng Zhang 0002, Weizhe Chen 0001, Zeren Huang, Minne Li, Yaodong Yang 0001, Weinan Zhang 0001, Jun Wang 0012 |
AAAI | 7 |
| 2020 | 3rd International Workshop on EntitY Retrieval and lEarning (EYRE 2020)abstractEntity retrieval has received increasing research attention. The recent progress in deep and machine learning techniques provides powerful tools for developing effective entity-centered solutions. This workshop series provides a platform where interdisciplinary studies of entity retrieval and learning can be presented, and focused discussions can take place. We also organize a shared task related to entity retrieval. The 3rd International Workshop on EntitY Retrieval and lEarning (EYRE 2020) was a half-day workshop co-located with the 29th ACM International Conference on Information and Knowledge Management (CIKM 2020) as a virtual event in Ireland. Gong Cheng 0001, Kalpa Gunaratna, Jun Wang 0012 |
CIKM | 3 |
| 2020 | U-rank: Utility-oriented Learning to Rank with Implicit FeedbackabstractLearning to rank with implicit feedback is one of the most important tasks in many real-world information systems where the objective is some specific utility, e.g., clicks and revenue. However, we point out that existing methods based on probabilistic ranking principle do not necessarily achieve the highest utility. To this end, we propose a novel ranking framework called U-rank that directly optimizes the expected utility of the ranking list. With a position-aware deep click-through rate prediction model, we address the attention bias considering both query-level and item-level features. Due to the item-specific attention bias modeling, the optimization for expected utility corresponds to a maximum weight matching on the item-position bipartite graph. We base the optimization of this objective in an efficient Lambdaloss framework, which is supported by both theoretical and empirical analysis. We conduct extensive experiments for both web search and recommender systems over three benchmark datasets and two proprietary datasets, where the performance gain of U-rank over state-of-the-arts is demonstrated. Moreover, our proposed U-rank has been deployed on a large-scale commercial recommender and a large improvement over the production baseline has been observed in an online A/B testing. Xinyi Dai, Qing Liu 0020, Yunjia Xi, Ruiming Tang, Weinan Zhang 0001, Xiuqiang He 0001, Jun Wang 0012, Yong Yu 0001 |
CIKM | 8 |
| 2020 | Learning to Infer User Hidden States for Online Sequential AdvertisingabstractTo drive purchase in online advertising, it is of the advertiser's great interest to optimize the sequential advertising strategy whose performance and interpretability are both important. The lack of interpretability in existing deep reinforcement learning methods makes it not easy to understand, diagnose and further optimize the strategy.In this paper, we propose our Deep Intents Sequential Advertising (DISA) method to address these issues. The key part of interpretability is to understand a consumer's purchase intent which is, however, unobservable (called hidden states). In this paper, we model this intention as a latent variable and formulate the problem as a Partially Observable Markov Decision Process (POMDP) where the underlying intents are inferred based on the observable behaviors. Large-scale industrial offline and online experiments demonstrate our method's superior performance over several baselines. The inferred hidden states are analyzed, and the results prove the rationality of our inference. Zhaoqing Peng, Junqi Jin, Yaodong Yang 0001, Rui Luo 0001, Jun Wang 0012, Weinan Zhang 0001, Chuan Yu 0002, Tiejian Luo, Han Li 0005, Jian Xu 0015, Kun Gai |
CIKM | 6 |
| 2020 | Off-Policy Reinforcement Learning for Efficient and Effective GAN Architecture Search
Yuan Tian 0014, Qin Wang 0013, Zhiwu Huang, Wen Li 0001, Dengxin Dai, Jun Wang 0012, Olga Fink |
ECCV (7) | 7 |
| 2020 | Multi-Agent Interactions Modeling with Correlated Policies
Minghuan Liu, Ming Zhou 0006, Weinan Zhang 0001, Yuzheng Zhuang, Jun Wang 0012, Wulong Liu, Yong Yu 0001 |
ICLR | 5 |
| 2020 | Multi-Agent Determinantal Q-LearningabstractCentralized training with decentralized execution has become an important paradigm in multi-agent learning. Though practical, current methods rely on restrictive assumptions to decompose the centralized value function across agents for execution. In this paper, we eliminate this restriction by proposing multi-agent determinantal Q-learning. Our method is established on Q-DPP, a novel extension of determinantal point process (DPP) to multi-agent setting. Q-DPP promotes agents to acquire diverse behavioral models; this allows a natural factorization of the joint Q-functions with no need for \emph{a priori} structural constraints on the value function or special network architectures. We demonstrate that Q-DPP generalizes major solutions including VDN, QMIX, and QTRAN on decentralizable cooperative tasks. To efficiently draw samples from Q-DPP, we develop a linear-time sampler with theoretical approximation guarantee. Our sampler also benefits exploration by coordinating agents to cover orthogonal directions in the state space during training. We evaluate our algorithm on multiple cooperative benchmarks; its effectiveness has been demonstrated when compared with the state-of-the-art. Yaodong Yang 0001, Ying Wen 0001, Jun Wang 0012, Kun Shao, David Mguni, Weinan Zhang 0001 |
ICML | 3 |
| 2020 | Modelling Bounded Rationality in Multi-Agent Interactions by Generalized Recursive ReasoningabstractThough limited in real-world decision making, most multi-agent reinforcement learning (MARL) models assume perfectly rational agents -- a property hardly met due to individual's cognitive limitation and/or the tractability of the decision problem. In this paper, we introduce generalized recursive reasoning (GR2) as a novel framework to model agents with different \emph{hierarchical} levels of rationality; our framework enables agents to exhibit varying levels of ``thinking'' ability thereby allowing higher-level agents to best respond to various less sophisticated learners. We contribute both theoretically and empirically. On the theory side, we devise the hierarchical framework of GR2 through probabilistic graphical models and prove the existence of a perfect Bayesian equilibrium. Within the GR2, we propose a practical actor-critic solver, and demonstrate its convergent property to a stationary point in two-player games through Lyapunov analysis. On the empirical side, we validate our findings on a variety of MARL benchmarks. Precisely, we first illustrate the hierarchical thinking process on the Keynes Beauty Contest, and then demonstrate significant improvements compared to state-of-the-art opponent modeling baselines on the normal-form games and the cooperative navigation benchmark. Ying Wen 0001, Yaodong Yang 0001, Jun Wang 0012 |
IJCAI | 3 |
| 2020 | Replica-Exchange Nosé-Hoover Dynamics for Bayesian Learning on Large Datasets
Rui Luo 0001, Qiang Zhang 0026, Yaodong Yang 0001, Jun Wang 0012 |
NeurIPS | 4 |
| 2020 | JIT2R: A Joint Framework for Item Tagging and Tag-based RecommendationabstractPredicting tags for a given item and leveraging tags to assist item recommendation are two popular research topics in the field of recommender system. Previous studies mostly focus only one of them to make contributions. However, we believe that these tasks are inherently correlated with each other: tags can provide additional information to profile items for more accurate recommendation; user behaviors can help to infer item relationships to benefit the item tagging process. In order to take the advantages of such mutually influential signals, we propose to integrate item tagging and tag-based recommendation into a unified model. We firstly design a basic framework, where the user-item interaction signals are leveraged to supervise the item tagging process. Then we extend the basic model with a bootstrapping technique to circulate such mutual improvements between different tasks. We conduct extensive experiments based on real-word datasets to demonstrate our model's superiorities. Xu Chen 0017, Changying Du, Xiuqiang He 0001, Jun Wang 0012 |
SIGIR | 4 |
| 2020 | A Deep Recurrent Survival Model for Unbiased RankingabstractPosition bias is a critical problem in information retrieval when dealing with implicit yet biased user feedback data. Unbiased ranking methods typically rely on causality models and debias the user feedback through inverse propensity weighting. While practical, these methods still suffer from two major problems. First, when infer a user click, the impact of the contextual information, such as documents that have been examined, is often ignored. Second, only the position bias is considered but other issues resulted from user browsing behaviors are overlooked. In this paper, we propose an end-to-end Deep Recurrent Survival Ranking (DRSR), a unified framework to jointly model user's various behaviors, to (i) consider the rich contextual information in the ranking list; and (ii) address the hidden issues underlying user behaviors, i.e., to mine observe pattern in queries without any click (non-click queries), and to model tracking logs which cannot truly reflect the user browsing intents (untrusted observation). Specifically, we adopt a recurrent neural network to model the contextual information and estimates the conditional likelihood of user feedback at each position. We then incorporate survival analysis techniques with the probability chain rule to mathematically recover the unbiased joint probability of one user's various behaviors. DRSR can be easily incorporated with both point-wise and pair-wise learning objectives. The extensive experiments over two large-scale industrial datasets demonstrate the significant performance gains of our model comparing with the state-of-the-arts. Jiarui Jin, Weinan Zhang 0001, Kan Ren, Guorui Zhou, Jian Xu 0015, Yong Yu 0001, Jun Wang 0012, Xiaoqiang Zhu, Kun Gai |
SIGIR | 8 |
| 2020 | GREASE: A Generative Model for Relevance Search over Knowledge GraphsabstractRelevance search is to find top-ranked entities in a knowledge graph (KG) that are relevant to a query entity. Relevance is ambiguous, particularly over a schema-rich KG like DBpedia which supports a wide range of different semantics of relevance based on numerous types of relations and attributes. As users may lack the expertise to formalize the desired semantics, supervised methods have emerged to learn the hidden user-defined relevance from user-provided examples. Along this line, in this paper we propose a novel generative model over KGs for relevance search, named GREASE. The model applies to meta-path based relevance where a meta-path characterizes a particular type of semantics of relating the query entity to answer entities. It is also extended to support properties that constrain answer entities. Extensive experiments on two large-scale KGs demonstrate that GREASE has advanced the state of the art in effectiveness, expressiveness, and efficiency. Tianshuo Zhou, Ziyang Li 0005, Gong Cheng 0001, Jun Wang 0012, Yuang Wei |
WSDM | 4 |
| 2020 | Strongly Constrained Discrete HashingabstractLearning to hash is a fundamental technique widely used in large-scale image retrieval. Most existing methods for learning to hash address the involved discrete optimization problem by the continuous relaxation of the binary constraint, which usually lead to large quantization errors and consequently suboptimal binary codes. A few discrete hashing methods have emerged recently. However, they either completely ignore some useful constraints (specifically the balance and decorrelation of hash bits) or just turn those constraints into regularizers that would make the optimization easier but less accurate. In this paper, we propose a novel supervised hashing method named, Strongly Constrained Discrete Hashing (SCDH), which overcomes such limitations. It can learn the binary codes for all examples in the training set, and meanwhile obtain a hash function for unseen samples with the above mentioned constraints preserved. Although the model of SCDH is fairly sophisticated, we are able to find closed-form solutions to all of its optimization subproblems and thus design an efficient algorithm that converges quickly. In addition, we extend SCDH to a kernelized version SCDHK. Our experiments on three large benchmark datasets have demonstrated that not only can SCDH and SCDHK achieve substantially higher MAP scores than state-of-the-art baselines, but they run much faster than those that are also supervised as well. Yong Chen 0008, Zhibao Tian, Hui Zhang 0028, Jun Wang 0012, Dell Zhang |
IEEE Trans. Image Process. | 4 |
| 2019 | Learning to Advertise for Organic Traffic Maximization in E-Commerce Product FeedsabstractMost e-commerce product feeds provide blended results of advertised products and recommended products to consumers. The underlying advertising and recommendation platforms share similar if not exactly the same set of candidate products. Consumers' behaviors on the advertised results constitute part of the recommendation model's training data and therefore can influence the recommended results. We refer to this process as Leverage. Considering this mechanism, we propose a novel perspective that advertisers can strategically bid through the advertising platform to optimize their recommended organic traffic. By analyzing the real-world data, we first explain the principles of Leverage mechanism, i.e., the dynamic models of Leverage. Then we introduce a novel Leverage optimization problem and formulate it with a Markov Decision Process. To deal with the sample complexity challenge in model-free reinforcement learning, we propose a novel Hybrid Training Leverage Bidding (HTLB) algorithm which combines the real-world samples and the emulator-generated samples to boost the learning speed and stability. Our offline experiments as well as the results from the online deployment demonstrate the superior performance of our approach. Dagui Chen, Junqi Jin, Weinan Zhang 0001, Lvyin Niu, Chuan Yu 0002, Jun Wang 0012, Han Li 0005, Jian Xu 0015, Kun Gai |
CIKM | 7 |
| 2019 | EYRE 2019: 2nd International Workshop on EntitY REtrievalabstractEntity retrieval has received increasing research attention from both the Information Retrieval (IR) and Semantic Web communities. This workshop series provides a platform where interdisciplinary studies of entity retrieval can be presented, and focused discussions can take place. We also organize two shared tasks related to entity retrieval. The 2nd International Workshop on EntitY REtrieval (EYRE 2019) was a half-day workshop co-located with the 28th ACM International Conference on Information and Knowledge Management (CIKM 2019) in Beijing, China. Gong Cheng 0001, Kalpa Gunaratna, Jun Wang 0012 |
CIKM | 3 |
| 2019 | CoRide: Joint Order Dispatching and Fleet Management for Multi-Scale Ride-Hailing PlatformsabstractHow to optimally dispatch orders to vehicles and how to trade off between immediate and future returns are fundamental questions for a typical ride-hailing platform. We model ride-hailing as a large-scale parallel ranking problem and study the joint decision-making task of order dispatching and fleet management in online ride-hailing platforms. This task brings unique challenges in the following four aspects. First, to facilitate a huge number of vehicles to act and learn efficiently and robustly, we treat each region cell as an agent and build a multi-agent reinforcement learning framework. Second, to coordinate the agents from different regions to achieve long-term benefits, we leverage the geographical hierarchy of the region grids to perform hierarchical reinforcement learning. Third, to deal with the heterogeneous and variant action space for joint order dispatching and fleet management, we design the action as the ranking weight vector to rank and select the specific order or the fleet management destination in a unified formulation. Fourth, to achieve the multi-scale ride-hailing platform, we conduct the decision-making process in a hierarchical way where a multi-head attention mechanism is utilized to incorporate the impacts of neighbor agents and capture the key agent in each scale. The whole novel framework is named as CoRide. Extensive experiments based on multiple cities real-world data as well as analytic synthetic data demonstrate that CoRide provides superior performance in terms of platform revenue and user experience in the task of city-wide hybrid order dispatching and fleet management over strong baselines. Jiarui Jin, Ming Zhou 0006, Weinan Zhang 0001, Minne Li, Zilong Guo, Zhiwei (Tony) Qin, Yan Jiao, Xiaocheng Tang, Jun Wang 0012, Guobin Wu 0001, Jieping Ye |
CIKM | 10 |
| 2019 | Learning Adaptive Display Exposure for Real-Time AdvertisingabstractIn E-commerce advertising, where product recommendations and product ads are presented to users simultaneously, the traditional setting is to display ads at fixed positions. However, under such a setting, the advertising system loses the flexibility to control the number and positions of ads, resulting in sub-optimal platform revenue and user experience. Consequently, major e-commerce platforms (e.g., Taobao.com) have begun to consider more flexible ways to display ads. In this paper, we investigate the problem of advertising with adaptive exposure: can we dynamically determine the number and positions of ads for each user visit under certain business constraints so that the platform revenue can be increased? More specifically, we consider two types of constraints: request-level constraint ensures user experience for each user visit, and platform-level constraint controls the overall platform monetization rate. We model this problem as a Constrained Markov Decision Process with per-state constraint (psCMDP) and propose a constrained two-level reinforcement learning approach to decompose the original problem into two relatively independent sub-problems. To accelerate policy learning, we also devise a constrained hindsight experience replay mechanism. Experimental evaluations on industry-scale real-world datasets demonstrate the merits of our approach in both obtaining higher revenue under the constraints and the effectiveness of the constrained hindsight experience replay mechanism. Weixun Wang, Junqi Jin, Jianye Hao, Chunjie Chen 0004, Chuan Yu 0002, Weinan Zhang 0001, Jun Wang 0012, Xiaotian Hao, Yixi Wang 0003, Han Li 0005, Jian Xu 0015, Kun Gai |
CIKM | 7 |
| 2019 | MarlRank: Multi-agent Reinforced Learning to RankabstractWhen estimating the relevancy between a query and a document, ranking models largely neglect the mutual information among documents. A common wisdom is that if two documents are similar in terms of the same query, they are more likely to have similar relevance score. To mitigate this problem, in this paper, we propose a multi-agent reinforced ranking model, named MarlRank. In particular, by considering each document as an agent, we formulate the ranking process as a multi-agent Markov Decision Process (MDP), where the mutual interactions among documents are incorporated in the ranking process. To compute the ranking list, each document predicts its relevance to a query considering not only its own query-document features but also its similar documents' features and actions. By defining reward as a function of NDCG, we can optimize our model directly on the ranking performance measure. Our experimental results on two LETOR benchmark datasets show that our model has significant performance gains over the state-of-art baselines. We also find that the NDCG shows an overall increasing trend along with the step of interactions, which demonstrates that the mutual information among documents helps improve the ranking performance. Shihao Zou, Mohammad Akbari 0001, Jun Wang 0012, Peng Zhang 0002 |
CIKM | 4 |
| 2019 | Factorized Q-learning for large-scale multi-agent systemsabstractDeep Q-learning has achieved significant success in single-agent decision making tasks. However, it is challenging to extend Q-learning to large-scale multi-agent scenarios, due to the explosion of action space resulting from the complex dynamics between the environment and the agents. In this paper, we propose to make the computation of multi-agent Q-learning tractable by treating the Q-function (w.r.t. state and joint-action) as a high-order high-dimensional tensor and then approximate it with factorized pairwise interactions. Furthermore, we utilize a composite deep neural network architecture for computing the factorized Q-function, share the model parameters among all the agents within the same group, and estimate the agents' optimal joint actions through a coordinate descent type algorithm. All these simplifications greatly reduce the model complexity and accelerate the learning process. Extensive experiments on two different multi-agent problems demonstrate the performance gain of our proposed approach in comparison with strong baselines, particularly when there are a large number of agents. Ming Zhou 0006, Yong Chen 0008, Ying Wen 0001, Yaodong Yang 0001, Yufeng Su, Weinan Zhang 0001, Dell Zhang, Jun Wang 0012 |
DAI | 8 |
| 2019 | Modeling User Return Time Using Inhomogeneous Poisson Process
Mohammad Akbari 0001, Alberto Cetoli, Stefano Bragaglia, Andrew D. O'Harney, Marc Sloan, Jun Wang 0012 |
ECIR (2) | 6 |
| 2019 | Probabilistic Recursive Reasoning for Multi-Agent Reinforcement Learning
Ying Wen 0001, Yaodong Yang 0001, Rui Luo 0001, Jun Wang 0012, Wei Pan 0004 |
ICLR (Poster) | 4 |
| 2019 | BayesNAS: A Bayesian Approach for Neural Architecture SearchabstractOne-Shot Neural Architecture Search (NAS) is a promising method to significantly reduce search time without any separate training. It can be treated as a Network Compression problem on the architecture parameters from an over-parameterized network. However, there are two issues associated with most one-shot NAS methods. First, dependencies between a node and its predecessors and successors are often disregarded which result in improper treatment over zero operations. Second, architecture parameters pruning based on their magnitude is questionable. In this paper, we employ the classic Bayesian learning approach to alleviate these two issues by modeling architecture parameters using hierarchical automatic relevance determination (HARD) priors. Unlike other NAS methods, we train the over-parameterized network for only one epoch then update the architecture. Impressively, this enabled us to find the architecture in both proxy and proxyless tasks on CIFAR-10 within only 0.2 GPU days using a single GPU. As a byproduct, our approach can be transferred directly to compress convolutional neural networks by enforcing structural sparsity which achieves extremely sparse networks without accuracy deterioration. Hongpeng Zhou, Jun Wang 0012, Wei Pan 0004 |
ICML | 3 |
| 2019 | A Regularized Opponent Model with Maximum Entropy ObjectiveabstractIn a single-agent setting, reinforcement learning (RL) tasks can be cast into an inference problem by introducing a binary random variable o, which stands for the "optimality". In this paper, we redefine the binary random variable o in multi-agent setting and formalize multi-agent reinforcement learning (MARL) as probabilistic inference. We derive a variational lower bound of the likelihood of achieving the optimality and name it as Regularized Opponent Model with Maximum Entropy Objective (ROMMEO). From ROMMEO, we present a novel perspective on opponent modeling and show how it can improve the performance of training agents theoretically and empirically in cooperative games. To optimize ROMMEO, we first introduce a tabular Q-iteration method ROMMEO-Q with proof of convergence. We extend the exact algorithm to complex environments by proposing an approximate version, ROMMEO-AC. We evaluate these two algorithms on the challenging iterated matrix game and differential game respectively and show that they can outperform strong MARL baselines. Zheng Tian 0002, Ying Wen 0001, Zhichen Gong, Faiz Punakkath, Shihao Zou, Jun Wang 0012 |
IJCAI | 6 |
| 2019 | Multi-View Reinforcement LearningabstractThis paper is concerned with multi-view reinforcement learning (MVRL), which allows for decision making when agents share common dynamics but adhere to different observation models. We define the MVRL framework by extending partially observable Markov decision processes (POMDPs) to support more than one observation model and propose two solution methods through observation augmentation and cross-view policy transfer. We empirically evaluate our method and demonstrate its effectiveness in a variety of environments. Specifically, we show reductions in sample complexities and computational time for acquiring policies that handle multi-view environments. Minne Li, Lisheng Wu, Jun Wang 0012, Haitham Bou-Ammar |
NeurIPS | 3 |
| 2019 | Efficient Ridesharing Order Dispatching with Mean Field Multi-Agent Reinforcement LearningabstractA fundamental question in any peer-to-peer ridesharing system is how to, both effectively and efficiently, dispatch user's ride requests to the right driver in real time. Traditional rule-based solutions usually work on a simplified problem setting, which requires a sophisticated hand-crafted weight design for either centralized authority control or decentralized multi-agent scheduling systems. Although recent approaches have used reinforcement learning to provide centralized combinatorial optimization algorithms with informative weight values, their single-agent setting can hardly model the complex interactions between drivers and orders. In this paper, we address the order dispatching problem using multi-agent reinforcement learning (MARL), which follows the distributed nature of the peer-to-peer ridesharing problem and possesses the ability to capture the stochastic demand-supply dynamics in large-scale ridesharing scenarios. Being more reliable than centralized approaches, our proposed MARL solutions could also support fully distributed execution through recent advances in the Internet of Vehicles (IoV) and the Vehicle-to-Network (V2N). Furthermore, we adopt the mean field approximation to simplify the local interactions by taking an average action among neighborhoods. The mean field approximation is capable of globally capturing dynamic demand-supply variations by propagating many local interactions between agents and the environment. Our extensive experiments have shown the significant improvements of MARL order dispatching algorithms over several strong baselines on the accumulated driver income (ADI), and order response rate measures. Besides, the simulated experiments with real data have also justified that our solution can alleviate the supply-demand gap during the rush hours, thus possessing the capability of reducing traffic congestion. Minne Li, Zhiwei (Tony) Qin, Yan Jiao, Yaodong Yang 0001, Jun Wang 0012, Guobin Wu 0001, Jieping Ye |
WWW | 5 |
| 2018 | Efficient Architecture Search by Network TransformationabstractTechniques for automatically designing deep neural network architectures such as reinforcement learning based approaches have recently shown promising results. However, their success is based on vast computational resources (e.g. hundreds of GPUs), making them difficult to be widely used. A noticeable limitation is that they still design and train each network from scratch during the exploration of the architecture space, which is highly inefficient. In this paper, we propose a new framework toward efficient architecture search by exploring the architecture space based on the current network and reusing its weights. We employ a reinforcement learning agent as the meta-controller, whose action is to grow the network depth or layer width with function-preserving transformations. As such, the previously validated networks can be reused for further exploration, thus saves a large amount of computational cost. We apply our method to explore the architecture space of the plain convolutional neural networks (no skip-connections, branching etc.) on image benchmark datasets (CIFAR-10, SVHN) with restricted computational resources (5 GPUs). Our method can design highly competitive networks that outperform existing networks using the same design scheme. On CIFAR-10, our model without skip-connections achieves 4.23% test error rate, exceeding a vast majority of modern architectures and approaching DenseNet. Furthermore, by applying our method to explore the DenseNet architecture space, we are able to achieve more accurate networks with fewer parameters. Han Cai, Tianyao Chen, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
AAAI | 5 |
| 2018 | Long Text Generation via Adversarial Training with Leaked InformationabstractAutomatically generating coherent and semantically meaningful text has many applications in machine translation, dialogue systems, image captioning, etc. Recently, by combining with policy gradient, Generative Adversarial Nets(GAN) that use a discriminative model to guide the training of the generative model as a reinforcement learning policy has shown promising results in text generation. However, the scalar guiding signal is only available after the entire text has been generated and lacks intermediate information about text structure during the generative process. As such, it limits its success when the length of the generated text samples is long (more than 20 words). In this paper, we propose a new framework, called LeakGAN, to address the problem for long text generation. We allow the discriminative net to leak its own high-level extracted features to the generative net to further help the guidance. The generator incorporates such informative signals into all generation steps through an additional MANAGER module, which takes the extracted features of current generated words and outputs a latent vector to guide the WORKER module for next-word generation.Our extensive experiments on synthetic data and various real-world tasks with Turing test demonstrate that LeakGAN is highly effective in long text generation and also improves the performance in short text generation scenarios. More importantly, without any supervision, LeakGAN would be able to implicitly learn sentence structures only through the interaction between MANAGER and WORKER. Jiaxian Guo, Sidi Lu, Han Cai, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
AAAI | 6 |
| 2018 | A Neural Stochastic Volatility ModelabstractIn this paper, we show that the recent integration of statistical models with deep recurrent neural networks provides a new way of formulating volatility (the degree of variation of time series) models that have been widely used in time series analysis and prediction in finance. The model comprises a pair of complementary stochastic recurrent neural networks: the generative network models the joint distribution of the stochastic volatility process; the inference network approximates the conditional distribution of the latent variables given the observables. Our focus here is on the formulation of temporal dynamics of volatility over time under a stochastic recurrent neural network framework. Experiments on real-world stock price datasets demonstrate that the proposed model generates a better volatility estimation and prediction that outperforms mainstream methods, e.g., deterministic models such as GARCH and its variants, and stochastic models namely the MCMC-based stochvol as well as the Gaussian-process-based, on average negative log-likelihood. Rui Luo 0001, Weinan Zhang 0001, Jun Wang 0012 |
AAAI | 4 |
| 2018 | MAgent: A Many-Agent Reinforcement Learning Platform for Artificial Collective IntelligenceabstractWe introduce MAgent, a platform to support research and development of many-agent reinforcement learning. Unlike previous research platforms on single or multi-agent reinforcement learning, MAgent focuses on supporting the tasks and the applications that require hundreds to millions of agents. Within the interactions among a population of agents, it enables not only the study of learning algorithms for agents' optimal polices, but more importantly, the observation and understanding of individual agent's behaviors and social phenomena emerging from the AI society, including communication languages, leaderships, altruism. MAgent is highly scalable and can host up to one million agents on a single GPU server. MAgent also provides flexible configurations for AI researchers to design their customized environments and agents. In this demo, we present three environments designed on MAgent and show emerged collective intelligence by learning from scratch. Lianmin Zheng, Han Cai, Ming Zhou 0006, Weinan Zhang 0001, Jun Wang 0012, Yong Yu 0001 |
AAAI | 6 |
| 2018 | Real-Time Bidding with Multi-Agent Reinforcement Learning in Display AdvertisingabstractReal-time advertising allows advertisers to bid for each impression for a visiting user. To optimize specific goals such as maximizing revenue and return on investment (ROI) led by ad placements, advertisers not only need to estimate the relevance between the ads and user's interests, but most importantly require a strategic response with respect to other advertisers bidding in the market. In this paper, we formulate bidding optimization with multi-agent reinforcement learning. To deal with a large number of advertisers, we propose a clustering method and assign each cluster with a strategic bidding agent. A practical Distributed Coordinated Multi-Agent Bidding (DCMAB) has been proposed and implemented to balance the tradeoff between the competition and cooperation among advertisers. The empirical study on our industry-scaled real-world data has demonstrated the effectiveness of our methods. Our results show cluster-based bidding would largely outperform single-agent and bandit approaches, and the coordinated bidding achieves better overall objectives than purely self-interested bidding agents. Junqi Jin, Chengru Song, Han Li 0005, Kun Gai, Jun Wang 0012, Weinan Zhang 0001 |
CIKM | 5 |
| 2018 | Learning Multi-touch Conversion Attribution with Dual-attention Mechanisms for Online AdvertisingabstractIn online advertising, the Internet users may be exposed to a sequence of different ad campaigns, i.e., display ads, search, or referrals from multiple channels, before led up to any final sales conversion and transaction. For both campaigners and publishers, it is fundamentally critical to estimate the contribution from ad campaign touch-points during the customer journey (conversion funnel) and assign the right credit to the right ad exposure accordingly. However, the existing research on the multi-touch attribution problem lacks a principled way of utilizing the users' pre-conversion actions (i.e., clicks), and quite often fails to model the sequential patterns among the touch points from a user's behavior data. To make it worse, the current industry practice is merely employing a set of arbitrary rules as the attribution model, e.g., the popular last-touch model assigns 100% credit to the final touch-point regardless of actual attributions. In this paper, we propose a Dual-attention Recurrent Neural Network (DARNN) for the multi-touch attribution problem. It learns the attribution values through an attention mechanism directly from the conversion estimation objective. To achieve this, we utilize sequence-to-sequence prediction for user clicks, and combine both post-view and post-click attribution patterns together for the final conversion estimation. To quantitatively benchmark attribution models, we also propose a novel yet practical attribution evaluation scheme through the proxy of budget allocation (under the estimated attributions) over ad channels. The experimental results on two real datasets demonstrate the significant performance gains of our attribution model against the state of the art. Kan Ren, Weinan Zhang 0001, Shuhao Liu 0002, Ya Zhang 0002, Yong Yu 0001, Jun Wang 0012 |
CIKM | 8 |
| 2018 | Activation Maximization Generative Adversarial Nets
Zhiming Zhou 0001, Han Cai, Shu Rong, Yuxuan Song 0002, Kan Ren, Weinan Zhang 0001, Jun Wang 0012, Yong Yu 0001 |
ICLR (Poster) | 7 |
| 2018 | Mean Field Multi-Agent Reinforcement LearningabstractExisting multi-agent reinforcement learning methods are limited typically to a small number of agents. When the agent number increases largely, the learning becomes intractable due to the curse of the dimensionality and the exponential growth of agent interactions. In this paper, we present Mean Field Reinforcement Learning where the interactions within the population of agents are approximated by those between a single agent and the average effect from the overall population or neighboring agents; the interplay between the two entities is mutually reinforced: the learning of the individual agent’s optimal policy depends on the dynamics of the population, while the dynamics of the population change according to the collective patterns of the individual policies. We develop practical mean field Q-learning and mean field Actor-Critic algorithms and analyze the convergence of the solution to Nash equilibrium. Experiments on Gaussian squeeze, Ising model, and battle games justify the learning effectiveness of our mean field approaches. In addition, we report the first result to solve the Ising model via model-free reinforcement learning methods. Yaodong Yang 0001, Rui Luo 0001, Minne Li, Ming Zhou 0006, Weinan Zhang 0001, Jun Wang 0012 |
ICML | 6 |
| 2018 | Learning to Design Games: Strategic Environments in Reinforcement LearningabstractIn typical reinforcement learning (RL), the environment is assumed given and the goal of the learning is to identify an optimal policy for the agent taking actions through its interactions with the environment. In this paper, we extend this setting by considering the environment is not given, but controllable and learnable through its interaction with the agent at the same time. This extension is motivated by environment design scenarios in the real-world, including game design, shopping space design and traffic signal design. Theoretically, we find a dual Markov decision process (MDP) w.r.t. the environment to that w.r.t. the agent, and derive a policy gradient solution to optimizing the parametrized environment. Furthermore, discontinuous environments are addressed by a proposed general generative framework. Our experiments on a Maze game design task show the effectiveness of the proposed algorithms in generating diverse and challenging Mazes against various agent settings. Haifeng Zhang 0002, Jun Wang 0012, Zhiming Zhou 0001, Weinan Zhang 0001, Yong Yu 0001, Wenxin Li 0005 |
IJCAI | 2 |
| 2018 | Explanation Mining: Post Hoc Interpretability of Latent Factor Models for Recommendation SystemsabstractThe widescale use of machine learning algorithms to drive decision-making has highlighted the critical importance of ensuring the interpretability of such models in order to engender trust in their output. The state-of-the-art recommendation systems use black-box latent factor models that provide no explanation of why a recommendation has been made, as they abstract their decision processes to a high-dimensional latent space which is beyond the direct comprehension of humans. We propose a novel approach for extracting explanations from latent factor recommendation systems by training association rules on the output of a matrix factorisation black-box model. By taking advantage of the interpretable structure of association rules, we demonstrate that predictive accuracy of the recommendation model can be maintained whilst yielding explanations with high fidelity to the black-box model on a unique industry dataset. Our approach mitigates the accuracy-interpretability trade-off whilst avoiding the need to sacrifice flexibility or use external data sources. We also contribute to the ill-defined problem of evaluating interpretability. Georgina Peake, Jun Wang 0012 |
KDD | 2 |
| 2018 | Thermostat-assisted continuously-tempered Hamiltonian Monte Carlo for Bayesian learningabstractIn this paper, we propose a novel sampling method, the thermostat-assisted continuously-tempered Hamiltonian Monte Carlo, for the purpose of multimodal Bayesian learning. It simulates a noisy dynamical system by incorporating both a continuously-varying tempering variable and the Nos\'e-Hoover thermostats. A significant benefit is that it is not only able to efficiently generate i.i.d. samples when the underlying posterior distributions are multimodal, but also capable of adaptively neutralising the noise arising from the use of mini-batches. While the properties of the approach have been studied using synthetic datasets, our experiments on three real datasets have also shown its performance gains over several strong baselines for Bayesian learning with various types of neural networks plunged in. Rui Luo 0001, Yaodong Yang 0001, Jun Wang 0012, Zhanxing Zhu |
NeurIPS | 4 |
| 2018 | Texygen: A Benchmarking Platform for Text Generation ModelsabstractWe introduce Texygen, a benchmarking platform to support research on open-domain text generation models. Texygen has not only implemented a majority of text generation models, but also covered a set of metrics that evaluate the diversity, the quality and the consistency of the generated texts. The Texygen platform could help standardize the research on text generation and improve the reproductivity and reliability of future research work in text generation. Yaoming Zhu, Sidi Lu, Lei Zheng 0004, Jiaxian Guo, Weinan Zhang 0001, Jun Wang 0012, Yong Yu 0001 |
SIGIR | 6 |
| 2018 | DEBTS: Delay Energy Balanced Task Scheduling in Homogeneous Fog NetworksabstractVehicular ad hoc networks, wireless sensor networks, Internet of Things, and mobile device-to-device communications can be modeled as different homogeneous fog networks, wherein similar terminals/things/devices/nodes are sharing their computation, communication, and storage resources in the neighborhood for achieving better system performance through effective collaborations. It is very desirable, but quite challenging, to simultaneously reduce service delay and energy consumption in such networks for delay-sensitive and energyconstraint applications, e.g., virtual reality and online 3-D gaming on mobile devices. In this paper, a cross-layer analytical framework is developed to formulate and study the balance between service delay and energy consumption. An effective control parameter V is derived to characterize their tradeoff relationship during dynamic task scheduling processes in fog networks. Combining this analysis with Lyapunov optimization techniques, a novel delay energy balanced tasking scheduling (DEBTS) algorithm is proposed to minimize the overall energy consumption while reducing average service delay and delay jitter. It is proved that DEBTS can achieve the theoretical [O(1/V), O(V)] tradeoff between these two performance metrics. Further, extensive simulation results show that DEBTS can offer much better delay-energy performance in task scheduling challenges. Specifically, for a typical V value of 4 × 104, DEBTS can save 26% and 29% more energy, and at the same time, reduce average service delay by 29% and 32%, than traditional random scheduling and least busy scheduling algorithms, respectively. Yang Yang 0001, Wuxiong Zhang, Yu Chen 0006, Xiliang Luo, Jun Wang 0012 |
IEEE Internet Things J. | 6 |
| 2018 | Bidding Machine: Learning to Bid for Directly Optimizing Profits in Display AdvertisingabstractReal-time bidding (RTB) based display advertising has become one of the key technological advances in computational advertising. RTB enables advertisers to buy individual ad impressions via an auction in real-time and facilitates the evaluation and the bidding of individual impressions across multiple advertisers. In RTB, the advertisers face three main challenges when optimizing their bidding strategies, namely (i) estimating the utility (e.g., conversions, clicks) of the ad impression, (ii) forecasting the market value (thus the cost) of the given ad impression, and (iii) deciding the optimal bid for the given auction based on the first two. Previous solutions assume the first two are solved before addressing the bid optimization problem. However, these challenges are strongly correlated and dealing with any individual problem independently may not be globally optimal. In this paper, we propose Bidding Machine, a comprehensive learning to bid framework, which consists of three optimizers dealing with each challenge above, and as a whole, jointly optimizes these three parts. We show that such a joint optimization would largely increase the campaign effectiveness and the profit. From the learning perspective, we show that the bidding machine can be updated smoothly with both offline periodical batch or online sequential training schemes. Our extensive offline empirical study and online A/B testing verify the high effectiveness of the proposed bidding machine. Kan Ren, Weinan Zhang 0001, Ke Chang, Yifei Rong, Yong Yu 0001, Jun Wang 0012 |
IEEE Trans. Knowl. Data Eng. | 6 |
| 2017 | SeqGAN: Sequence Generative Adversarial Nets with Policy GradientabstractAs a new way of training generative models, Generative Adversarial Net (GAN) that uses a discriminative model to guide the training of the generative model has enjoyed considerable success in generating real-valued data. However, it has limitations when the goal is for generating sequences of discrete tokens. A major reason lies in that the discrete outputs from the generative model make it difficult to pass the gradient update from the discriminative model to the generative model. Also, the discriminative model can only assess a complete sequence, while for a partially generated sequence, it is non-trivial to balance its current score and the future one once the entire sequence has been generated. In this paper, we propose a sequence generation framework, called SeqGAN, to solve the problems. Modeling the data generator as a stochastic policy in reinforcement learning (RL), SeqGAN bypasses the generator differentiation problem by directly performing gradient policy update. The RL reward signal comes from the GAN discriminator judged on a complete sequence, and is passed back to the intermediate state-action steps using Monte Carlo search. Extensive experiments on synthetic data and real-world tasks demonstrate significant improvements over strong baselines. Lantao Yu, Weinan Zhang 0001, Jun Wang 0012, Yong Yu 0001 |
AAAI | 3 |
| 2017 | Content Recommendation by Noise Contrastive Transfer Learning of Feature RepresentationabstractPersonalized recommendation has been proved effective as a content discovery tool for many online news publishers. As fresh news articles are frequently coming to the system while the old ones are fading away quickly, building a consistent and coherent feature representation over the ever-changing articles pool is fundamental to the performance of the recommendation. However, learning a good feature representation is challenging, especially for some small publishers that have normally fewer than 10,000 articles each year. In this paper, we consider to transfer knowledge from a larger text corpus. In our proposed solution, an effective article recommendation engine can be established with a small number of target publisher articles by transferring knowledge from a large corpus of text with a different distribution. Specifically, we leverage noise contrastive estimation techniques to learn the word conditional distribution given the context words, where the noise conditional distribution is pre-trained from the large corpus. Our solution has been deployed in a commercial recommendation service. The large-scale online A/B testing on two commercial publishers demonstrates up to 9.97% relative overall performance gain of our proposed model on the recommendation click-though rate metric over the non-transfer learning baselines. Guanyu Tao, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
CIKM | 5 |
| 2017 | Dynamic Attention Deep Model for Article Recommendation by Learning Human Editors' DemonstrationabstractAs aggregators, online news portals face great challenges in continuously selecting a pool of candidate articles to be shown to their users. Typically, those candidate articles are recommended manually by platform editors from a much larger pool of articles aggregated from multiple sources. Such a hand-pick process is labor intensive and time-consuming. In this paper, we study the editor article selection behavior and propose a learning by demonstration system to automatically select a subset of articles from the large pool. Our data analysis shows that (i) editors' selection criteria are non-explicit, which are less based only on the keywords or topics, but more depend on the quality and attractiveness of the writing from the candidate article, which is hard to capture based on traditional bag-of-words article representation. And (ii) editors' article selection behaviors are dynamic: articles with different data distribution come into the pool everyday and the editors' preference varies, which are driven by some underlying periodic or occasional patterns. To address such problems, we propose a meta-attention model across multiple deep neural nets to (i) automatically catch the editors' underlying selection criteria via the automatic representation learning of each article and its interaction with the meta data and (ii) adaptively capture the change of such criteria via a hybrid attention model. The attention model strategically incorporates multiple prediction models, which are trained in previous days. The system has been deployed in a commercial article feed platform. A 9-day A/B testing has demonstrated the consistent superiority of our proposed model over several strong baselines. Xuejian Wang, Lantao Yu, Kan Ren, Guanyu Tao, Weinan Zhang 0001, Yong Yu 0001, Jun Wang 0012 |
KDD | 7 |
| 2017 | IRGAN: A Minimax Game for Unifying Generative and Discriminative Information Retrieval ModelsabstractThis paper provides a unified account of two schools of thinking in information retrieval modelling: the generative retrieval focusing on predicting relevant documents given a query, and the discriminative retrieval focusing on predicting relevancy given a query-document pair. We propose a game theoretical minimax game to iteratively optimise both models. On one hand, the discriminative model, aiming to mine signals from labelled and unlabelled data, provides guidance to train the generative model towards fitting the underlying relevance distribution over documents given the query. On the other hand, the generative model, acting as an attacker to the current discriminative model, generates difficult examples for the discriminative model in an adversarial way by minimising its discrimination objective. With the competition between these two models, we show that the unified framework takes advantage of both schools of thinking: (i) the generative model learns to fit the relevance distribution over documents via the signals from the discriminative model, and (ii) the discriminative model is able to exploit the unlabelled data selected by the generative model to achieve a better estimation for document ranking. Our experimental results have demonstrated significant performance gains as much as 23.96% on [email protected] and 15.50% on MAP over strong baselines in a variety of applications including web search, item recommendation, and question answering. Jun Wang 0012, Lantao Yu, Weinan Zhang 0001, Benyou Wang, Peng Zhang 0002, Dell Zhang |
SIGIR | 1 |
| 2017 | Real-Time Bidding by Reinforcement Learning in Display AdvertisingabstractThe majority of online display ads are served through real-time bidding (RTB) --- each ad display impression is auctioned off in real-time when it is just being generated from a user visit. To place an ad automatically and optimally, it is critical for advertisers to devise a learning algorithm to cleverly bid an ad impression in real-time. Most previous works consider the bid decision as a static optimization problem of either treating the value of each impression independently or setting a bid price to each segment of ad volume. However, the bidding for a given ad campaign would repeatedly happen during its life span before the budget runs out. As such, each bid is strategically correlated by the constrained budget and the overall effectiveness of the campaign (e.g., the rewards from generated clicks), which is only observed after the campaign has completed. Thus, it is of great interest to devise an optimal bidding strategy sequentially so that the campaign budget can be dynamically allocated across all the available impressions on the basis of both the immediate and future rewards. In this paper, we formulate the bid decision process as a reinforcement learning problem, where the state space is represented by the auction information and the campaign's real-time parameters, while an action is the bid price to set. By modeling the state transition via auction competition, we build a Markov Decision Process framework for learning the optimal bidding policy to optimize the advertising performance in the dynamic real-time bidding environment. Furthermore, the scalability problem from the large real-world auction volume and campaign budget is well handled by state value approximation using neural networks. The empirical study on two large-scale real-world datasets and the live A/B testing on a commercial platform have demonstrated the superior performance and high efficiency compared to state-of-the-art methods. Han Cai, Kan Ren, Weinan Zhang 0001, Kleanthis Malialis, Jun Wang 0012, Yong Yu 0001, Defeng Guo |
WSDM | 5 |
| 2017 | Managing Risk of Bidding in Display AdvertisingabstractIn this paper, we deal with the uncertainty of bidding for display advertising. Similar to the financial market trading, real-time bidding (RTB) based display advertising employs an auction mechanism to automate the impression level media buying; and running a campaign is no different than an investment of acquiring new customers in return for obtaining additional converted sales. Thus, how to optimally bid on an ad impression to drive the profit and return-on-investment becomes essential. However, the large randomness of the user behaviors and the cost uncertainty caused by the auction competition may result in a significant risk from the campaign performance estimation. In this paper, we explicitly model the uncertainty of user click-through rate estimation and auction competition to capture the risk. We borrow an idea from finance and derive the value at risk for each ad display opportunity. Our formulation results in two risk-aware bidding strategies that penalize risky ad impressions and focus more on the ones with higher expected return and lower risk. The empirical study on real-world data demonstrates the effectiveness of our proposed risk-aware bidding strategies: yielding profit gains of 15.4% in offline experiments and up to 17.5% in an online A/B test on a commercial RTB platform over the widely applied bidding strategies. Haifeng Zhang 0002, Weinan Zhang 0001, Yifei Rong, Kan Ren, Wenxin Li 0005, Jun Wang 0012 |
WSDM | 6 |
| 2016 | User Response Learning for Directly Optimizing Campaign Performance in Display AdvertisingabstractLearning and predicting user responses, such as clicks and conversions, are crucial for many Internet-based businesses including web search, e-commerce, and online advertising. Typically, a user response model is established by optimizing the prediction accuracy, e.g., minimizing the error between the prediction and the ground truth user response. However, in many practical cases, predicting user responses is only part of a rather larger predictive or optimization task, where on one hand, the accuracy of a user response prediction determines the final (expected) utility to be optimized, but on the other hand, its learning may also be influenced from the follow-up stochastic process. It is, thus, of great interest to optimize the entire process as a whole rather than treat them independently or sequentially. In this paper, we take real-time display advertising as an example, where the predicted user's ad click-through rate (CTR) is employed to calculate a bid for an ad impression in the second price auction. We reformulate a common logistic regression CTR model by putting it back into its subsequent bidding context: rather than minimizing the prediction error, the model parameters are learned directly by optimizing campaign profit. The gradient update resulted from our formulations naturally fine-tunes the cases where the market competition is high, leading to a more cost-effective bidding. Our experiments demonstrate that, while maintaining comparable CTR prediction accuracy, our proposed user response learning leads to campaign profit gains as much as 78.2% for offline test and 25.5% for online A/B test over strong baselines. Kan Ren, Weinan Zhang 0001, Yifei Rong, Haifeng Zhang 0002, Yong Yu 0001, Jun Wang 0012 |
CIKM | 6 |
| 2016 | Real-Time Bidding Based Display Advertising: Mechanisms and Algorithms
Jun Wang 0012, Shuai Yuan 0002, Weinan Zhang 0001 |
ECIR | 1 |
| 2016 | Implicit Look-Alike Modelling in Display Ads - Transfer Collaborative Filtering to CTR Estimation
Weinan Zhang 0001, Lingxi Chen, Jun Wang 0012 |
ECIR | 3 |
| 2016 | Deep Learning over Multi-field Categorical Data - - A Case Study on User Response Prediction
Weinan Zhang 0001, Jun Wang 0012 |
ECIR | 3 |
| 2016 | Product-Based Neural Networks for User Response PredictionabstractPredicting user responses, such as clicks and conversions, is of great importance and has found its usage inmany Web applications including recommender systems, websearch and online advertising. The data in those applicationsis mostly categorical and contains multiple fields, a typicalrepresentation is to transform it into a high-dimensional sparsebinary feature representation via one-hot encoding. Facing withthe extreme sparsity, traditional models may limit their capacityof mining shallow patterns from the data, i.e. low-order featurecombinations. Deep models like deep neural networks, on theother hand, cannot be directly applied for the high-dimensionalinput because of the huge feature space. In this paper, we proposea Product-based Neural Networks (PNN) with an embeddinglayer to learn a distributed representation of the categorical data, a product layer to capture interactive patterns between interfieldcategories, and further fully connected layers to explorehigh-order feature interactions. Our experimental results on twolarge-scale real-world ad click datasets demonstrate that PNNsconsistently outperform the state-of-the-art models on various metrics. Yanru Qu, Han Cai, Kan Ren, Weinan Zhang 0001, Yong Yu 0001, Ying Wen 0001, Jun Wang 0012 |
ICDM | 7 |
| 2016 | Bid-aware Gradient Descent for Unbiased Learning with Censored Data in Display AdvertisingabstractIn real-time display advertising, ad slots are sold per impression via an auction mechanism. For an advertiser, the campaign information is incomplete --- the user responses (e.g, clicks or conversions) and the market price of each ad impression are observed only if the advertiser's bid had won the corresponding ad auction. The predictions, such as bid landscape forecasting, click-through rate (CTR) estimation, and bid optimisation, are all operated in the pre-bid stage with full-volume bid request data. However, the training data is gathered in the post-bid stage with a strong bias towards the winning impressions. A common solution for learning over such censored data is to reweight data instances to correct the discrepancy between training and prediction. However, little study has been done on how to obtain the weights independent of previous bidding strategies and consequently integrate them into the final CTR prediction and bid generation steps. In this paper, we formulate CTR estimation and bid optimisation under such censored auction data. Derived from a survival model, we show that historic bid information is naturally incorporated to produce Bid-aware Gradient Descents (BGD) which controls both the importance and the direction of the gradient to achieve unbiased learning. The empirical study based on two large-scale real-world datasets demonstrates remarkable performance gains from our solution. The learning framework has been deployed on Yahoo!'s real-time bidding platform and provided 2.97% AUC lift for CTR estimation and 9.30% eCPC drop for bid optimisation in an online A/B test. Weinan Zhang 0001, Tianxiong Zhou, Jun Wang 0012, Jian Xu 0015 |
KDD | 3 |
| 2016 | Functional Bid Landscape Forecasting for Display Advertising
Kan Ren, Weinan Zhang 0001, Jun Wang 0012, Yong Yu 0001 |
ECML/PKDD (1) | 4 |
| 2016 | Bayesian Performance Comparison of Text ClassifiersabstractHow can we know whether one classifier is really better than the other? In the area of text classification, since the publication of Yang and Liu's seminal SIGIR-1999 paper, it has become a standard practice for researchers to apply null-hypothesis significance testing (NHST) on their experimental results in order to establish the superiority of a classifier. However, such a frequentist approach has a number of inherent deficiencies and limitations, e.g., the inability to accept the null hypothesis (that the two classifiers perform equally well), the difficulty to compare commonly-used multivariate performance measures like F1 scores instead of accuracy, and so on. In this paper, we propose a novel Bayesian approach to the performance comparison of text classifiers, and argue its advantages over the traditional frequentist approach based on t-test etc. In contrast to the existing probabilistic model for F1 scores which is unpaired, our proposed model takes the correlation between classifiers into account and thus achieves greater statistical power. Using several typical text classification algorithms and a benchmark dataset, we demonstrate that the our approach provides rich information about the difference between two classifiers' performances. Dell Zhang, Jun Wang 0012, Emine Yilmaz, Xiaoling Wang 0004 |
SIGIR | 2 |
| 2016 | Feedback Control of Real-Time Display AdvertisingabstractReal-Time Bidding (RTB) is revolutionising display advertising by facilitating per-impression auctions to buy ad impressions as they are being generated. Being able to use impression-level data, such as user cookies, encourages user behaviour targeting, and hence has significantly improved the effectiveness of ad campaigns. However, a fundamental drawback of RTB is its instability because the bid decision is made per impression and there are enormous fluctuations in campaigns' key performance indicators (KPIs). As such, advertisers face great difficulty in controlling their campaign performance against the associated costs. In this paper, we propose a feedback control mechanism for RTB which helps advertisers dynamically adjust the bids to effectively control the KPIs, e.g., the auction winning ratio and the effective cost per click. We further formulate an optimisation framework to show that the proposed feedback control mechanism also has the ability of optimising campaign performance. By settling the effective cost per click at an optimal reference value, the number of campaign's ad clicks can be maximised with the budget constraint. Our empirical study based on real-world data verifies the effectiveness and robustness of our RTB control system in various situations. The proposed feedback control mechanism has also been deployed on a commercial RTB platform and the online test has shown its success in generating controllable advertising performance. Weinan Zhang 0001, Yifei Rong, Jun Wang 0012, Tianchi Zhu, Xiao Fan Wang 0001 |
WSDM | 3 |
| 2015 | Document Boltzmann Machines for Information Retrieval
Peng Zhang 0002, Yuexian Hou, Dawei Song 0001, Jun Wang 0012 |
ECIR | 5 |
| 2015 | A Bayesian Hierarchical Model for Comparing Average F1 ScoresabstractIn multi-class text classification, the performance (effectiveness) of a classifier is usually measured by micro-averaged and macro-averaged F1 scores. However, the scores themselves do not tell us how reliable they are in terms of forecasting the classifier's future performance on unseen data. In this paper, we propose a novel approach to explicitly modelling the uncertainty of average F1 scores through Bayesian reasoning, and demonstrate that it can provide much more comprehensive performance comparison between text classifiers than the traditional frequentist null hypothesis significance testing (NHST). Dell Zhang, Jun Wang 0012, Xiaoling Wang 0004 |
ICDM | 2 |
| 2015 | Statistical Arbitrage Mining for Display AdvertisingabstractWe study and formulate arbitrage in display advertising. Real-Time Bidding (RTB) mimics stock spot exchanges and utilises computers to algorithmically buy display ads per impression via a real-time auction. Despite the new automation, the ad markets are still informationally inefficient due to the heavily fragmented marketplaces. Two display impressions with similar or identical effectiveness (e.g., measured by conversion or click-through rates for a targeted audience) may sell for quite different prices at different market segments or pricing schemes. In this paper, we propose a novel data mining paradigm called Statistical Arbitrage Mining (SAM) focusing on mining and exploiting price discrepancies between two pricing schemes. In essence, our SAMer is a meta-bidder that hedges advertisers' risk between CPA (cost per action)-based campaigns and CPM (cost per mille impressions)-based ad inventories; it statistically assesses the potential profit and cost for an incoming CPM bid request against a portfolio of CPA campaigns based on the estimated conversion rate, bid landscape and other statistics learned from historical data. In SAM, (i) functional optimisation is utilised to seek for optimal bidding to maximise the expected arbitrage net profit, and (ii) a portfolio-based risk management solution is leveraged to reallocate bid volume and budget across the set of campaigns to make a risk and return trade-off. We propose to jointly optimise both components in an EM fashion with high efficiency to help the meta-bidder successfully catch the transient statistical arbitrage opportunities in RTB. Both the offline experiments on a real-world large-scale dataset and online A/B tests on a commercial platform demonstrate the effectiveness of our proposed solution in exploiting arbitrage in various model settings and market environments. Weinan Zhang 0001, Jun Wang 0012 |
KDD | 2 |
| 2015 | Risk-Hedged Venture Capital Investment RecommendationabstractWith the increasing accessibility of transactional data in venture finance, venture capital firms (VCs) face great challenges in developing quantitative tools to identify new investment opportunities. Recommendation techniques have the possibility of helping VCs making data-driven investment decisions by providing an automatic screening process of a large number of startups across different domains on the basis of their past investment data. A previous study has shown the potential advantage of using collaborative filtering to catch and predict the VCs' investment behaviours. However, two fundamental challenges in venture finance make conventional recommendation techniques difficult to apply. First, risk factors should be cautiously considered when making investments: for a potential startup, a VC needs to specifically estimate how well this new investment can fit into its holding investment portfolio in such a way that investment risk can be hedged. Second, The investment behaviours are much sparser than conventional recommendation applications and a VC's investments are usually limited to a few industry categories, making it impossible to use a topic-diversification method to hedge the risk. In this paper, we solve the startup recommendation problem from a risk management perspective. We propose 5 risk-aware startup selection and ranking algorithms to catch the VCs' investment behaviours and predict their new investments. Apart from the contribution on the new risk-aware recommendation model, our experiments on the collected CrunchBase dataset show significant performance improvements over strong baselines. Weinan Zhang 0001, Jun Wang 0012 |
RecSys | 3 |
| 2015 | Real-Time Bidding: A New Frontier of Computational Advertising ResearchabstractIn display and mobile advertising, the most significant development in recent years is the Real-Time Bidding (RTB), which allows selling and buying in real-time one ad impression at a time. Since then, RTB has fundamentally changed the landscape of the digital marketing by scaling the buying process across a large number of available inventories. The demand for automation, integration and optimisation in RTB brings new research opportunities in the IR/DM/ML fields. However, despite its rapid growth and huge potential, many aspects of RTB remain unknown to the research community for many reasons. In this tutorial, together with invited distinguished speakers from online advertising industry, we aim to bring the insightful knowledge from the real-world systems to bridge the gaps and provide an overview of the fundamental infrastructure, algorithms, and technical and research challenges of this new frontier of computational advertising. We will also introduce to researchers the datasets, tools, and platforms which are publicly available thus they can get hands-on quickly. Jun Wang 0012, Shuai Yuan 0002 |
WSDM | 1 |
| 2015 | Dynamic Information Retrieval ModelingabstractIn Dynamic Information Retrieval modeling we model dynamic systems which change or adapt over time or a sequence of events using a range of techniques from artificial intelligence and reinforcement learning. Many of the open problems in current IR research can be described as dynamic systems, for instance, session search or computational advertising. State of the art research provides solutions to these problems that are responsive to a changing environment, learn from past interactions and predict future utility. Advances in IR interface, personalization and ad display demand models that can react to users in real time and in an intelligent, contextual way. The objective of this half-day tutorial is to provide a comprehensive and up-to-date introduction to Dynamic Information Retrieval Modeling. We motivate a conceptual model linking static, interactive and dynamic retrieval and use this to define dynamics within the context of IR. We then cover a number of algorithms and techniques from the artificial intelligence (AI) and online learning literature such as Markov Decision Processes (MDP), their partially observable variation (POMDP) and multi-armed bandits. Grace Hui Yang, Marc Sloan, Jun Wang 0012 |
WSDM | 3 |
| 2015 | Resource allocation algorithm based on hybrid particle swarm optimization for multiuser cognitive OFDM network
Lei Xu 0015, Jun Wang 0012, Qianmu Li, Xiaofei Zhang 0001 |
Expert Syst. Appl. | 2 |
| 2015 | A term-based methodology for query reformulation understanding
Marc Sloan, Grace Hui Yang, Jun Wang 0012 |
Inf. Retr. J. | 3 |
| 2015 | Multi-Keyword Multi-Click Advertisement Option Contracts for Sponsored SearchabstractIn sponsored search, advertisement (abbreviated ad) slots are usually sold by a search engine to an advertiser through an auction mechanism in which advertisers bid on keywords. In theory, auction mechanisms have many desirable economic properties. However, keyword auctions have a number of limitations including: the uncertainty in payment prices for advertisers; the volatility in the search engine’s revenue; and the weak loyalty between advertiser and search engine. In this article, we propose a special ad option that alleviates these problems. In our proposal, an advertiser can purchase an option from a search engine in advance by paying an upfront fee, known as the option price. The advertiser then has the right, but no obligation, to purchase among the prespecified set of keywords at the fixed cost-per-clicks (CPCs) for a specified number of clicks in a specified period of time. The proposed option is closely related to a special exotic option in finance that contains multiple underlying assets (multi-keyword) and is also multi-exercisable (multi-click). This novel structure has many benefits: advertisers can have reduced uncertainty in advertising; the search engine can improve the advertisers’ loyalty as well as obtain a stable and increased expected revenue over time. Since the proposed ad option can be implemented in conjunction with the existing keyword auctions, the option price and corresponding fixed CPCs must be set such that there is no arbitrage between the two markets. Option pricing methods are discussed and our experimental results validate the development. Compared to keyword auctions, a search engine can have an increased expected revenue by selling an ad option. Bowei Chen 0001, Jun Wang 0012, Ingemar J. Cox, Mohan Kankanhalli |
ACM Trans. Intell. Syst. Technol. | 2 |
| 2014 | Generalized Bias-Variance Evaluation of TREC Participated SystemsabstractRecent research has shown that the improvement of mean retrieval effectiveness (e.g., MAP) may sacrifice the retrieval stability across queries, implying a tradeoff between effectiveness and stability. The evaluation of both effectiveness and stability are often based on a baseline model, which could be weak or biased. In addition, the effectiveness-stability tradeoff has not been systematically or quantitatively evaluated over TREC participated systems. The above two problems, to some extent, limit our awareness of such tradeoff and its impact on developing future IR models. In this paper, motivated by a recently proposed bias-variance based evaluation, we adopt a strong and unbiased "baseline", which is a virtual target model constructed by the best performance (for each query) among all the participated systems in a retrieval task. We also propose generalized bias-variance metrics, based on which a systematic and quantitative evaluation of the effectiveness-stability tradeoff is carried out over the participated systems in the TREC Ad-hoc Track (1993-1999) and Web Track (2010-2012). We observe a clear effectiveness-stability tradeoff, with a trend of becoming more obvious in more recent years. This implies that when we pursue more effective IR systems over years, the stability has become problematic and could have been largely overlooked. Peng Zhang 0002, Linxue Hao, Dawei Song 0001, Jun Wang 0012, Yuexian Hou, Bin Hu 0001 |
CIKM | 4 |
| 2014 | An empirical study of reserve price optimisation in real-time biddingabstractIn this paper, we report the first empirical study and live test of the reserve price optimisation problem in the context of Real-Time Bidding (RTB) display advertising from an operational environment. A reserve price is the minimum that the auctioneer would accept from bidders in auctions, and in a second price auction it could potentially uplift the auctioneer's revenue by charging winners the reserve price instead of the second highest bids. As such it has been used for sponsored search and been well studied in that context. However, comparing with sponsored search and contextual advertising, this problem in the RTB context is less understood yet more critical for publishers because 1) bidders have to submit a bid for each individual impression, which mostly is associated with user data that is subject to change over time. This, coupled with practical constraints such as the budget, campaigns' life time, etc. makes the theoretical result from optimal auction theory not necessarily applicable and a further empirical study is required to confirm its optimality from the real-world system; 2) in RTB an advertiser is facing nearly unlimited supply and the auction is almost done in "last second", which encourages spending less on the high cost ad placements. This could imply the loss of bid volume over time if a correct reserve price is not in place. In this paper we empirically examine several commonly adopted algorithms for setting up a reserve price. We report our results of a large scale online experiment in a production platform. The results suggest the our proposed game theory based OneShot algorithm performed the best and the superiority is significant in most cases. Shuai Yuan 0002, Jun Wang 0012, Bowei Chen 0001, Peter Mason, Sam Seljan |
KDD | 2 |
| 2014 | Optimal real-time bidding for display advertisingabstractIn this paper we study bid optimisation for real-time bidding (RTB) based display advertising. RTB allows advertisers to bid on a display ad impression in real time when it is being generated. It goes beyond contextual advertising by motivating the bidding focused on user data and it is different from the sponsored search auction where the bid price is associated with keywords. For the demand side, a fundamental technical challenge is to automate the bidding process based on the budget, the campaign objective and various information gathered in runtime and in history. In this paper, the programmatic bidding is cast as a functional optimisation problem. Under certain dependency assumptions, we derive simple bidding functions that can be calculated in real time; our finding shows that the optimal bid has a non-linear relationship with the impression level evaluation such as the click-through rate and the conversion rate, which are estimated in real time from the impression level features. This is different from previous work that is mainly focused on a linear bidding function. Our mathematical derivation suggests that optimal bidding strategies should try to bid more impressions rather than focus on a small set of high valued impressions because according to the current RTB market data, compared to the higher evaluated impressions, the lower evaluated ones are more cost effective and the chances of winning them are relatively higher. Aside from the theoretical insights, offline experiments on a real dataset and online experiments on a production RTB system verify the effectiveness of our proposed optimal bidding strategies and the functional optimisation framework. Weinan Zhang 0001, Shuai Yuan 0002, Jun Wang 0012 |
KDD | 3 |
| 2014 | Dynamic information retrieval modelingabstractDynamic aspects of Information Retrieval (IR), including changes found in data, users and systems, are increasingly being utilized in search engines and information filtering systems. Existing IR techniques are limited in their ability to optimize over changes, learn with minimal computational footprint and be responsive and adaptive. The objective of this tutorial is to provide a comprehensive and up-to-date introduction to Dynamic Information Retrieval Modeling, the statistical modeling of IR systems that can adapt to change. It will cover techniques ranging from classic relevance feedback to the latest applications of partially observable Markov decision processes (POMDPs) and a handful of useful algorithms and tools for solving IR problems incorporating dynamics. Grace Hui Yang, Marc Sloan, Jun Wang 0012 |
SIGIR | 3 |
| 2014 | Bias-variance analysis in estimating true query model for information retrieval
Peng Zhang 0002, Dawei Song 0001, Jun Wang 0012, Yuexian Hou |
Inf. Process. Manag. | 3 |
| 2014 | Introduction to the Special Issue on Diversity and Discovery in Recommender Systemsabstractintroduction Share on Introduction to the Special Issue on Diversity and Discovery in Recommender Systems Authors: Pablo Castells Universidad Autónoma de Madrid Universidad Autónoma de MadridView Profile , Jun Wang University College London University College LondonView Profile , Rubén Lara Telefónica Digital Telefónica DigitalView Profile , Dell Zhang Birkbeck, University of London Birkbeck, University of LondonView Profile Authors Info & Claims ACM Transactions on Intelligent Systems and TechnologyVolume 5Issue 4January 2015 Article No.: 52pp 1–3https://doi.org/10.1145/2668113Online:15 December 2014Publication History 4citation289DownloadsMetricsTotal Citations4Total Downloads289Last 12 Months4Last 6 weeks2 Get Citation AlertsNew Citation Alert added!This alert has been successfully added and will be sent to:You will be notified whenever a record that you have chosen has been cited.To manage your alert preferences, click on the button below.Manage my Alerts New Citation Alert!Please log in to your account Save to BinderSave to BinderCreate a New BinderNameCancelCreateExport CitationPublisher SiteGet Access Pablo Castells, Jun Wang 0012, Rubén Lara, Dell Zhang |
ACM Trans. Intell. Syst. Technol. | 2 |
| 2013 | Interactive collaborative filteringabstractIn this paper, we study collaborative filtering (CF) in an interactive setting, in which a recommender system continuously recommends items to individual users and receives interactive feedback. Whilst users enjoy sequential recommendations, the recommendation predictions are constantly refined using up-to-date feedback on the recommended items. Bringing the interactive mechanism back to the CF process is fundamental because the ultimate goal for a recommender system is about the discovery of interesting items for individual users and yet users' personal preferences and contexts evolve over time during the interactions with the system. This requires us not to distinguish between the stages of collecting information to construct the user profile and making recommendations, but to seamlessly integrate these stages together during the interactive process, with the goal of maximizing the overall recommendation accuracy throughout the interactions. This mechanism naturally addresses the cold-start problem as any user can immediately receive sequential recommendations without providing ratings beforehand. We formulate the interactive CF with the probabilistic matrix factorization (PMF) framework, and leverage several exploitation-exploration algorithms to select items, including the empirical Thompson sampling and upper confidence bound based algorithms. We conduct our experiment on cold-start users as well as warm-start users with drifting taste. Results show that the proposed methods have significant improvements over several strong baselines for the MovieLens, EachMovie and Netflix datasets. Weinan Zhang 0001, Jun Wang 0012 |
CIKM | 3 |
| 2013 | To personalize or not: a risk management perspectiveabstractPersonalization techniques have been widely adopted in many recommender systems. However, experiments on real-world datasets show that for some users in certain contexts, personalized recommendations do not necessarily perform better than recommendations that rely purely on popularity. Broadly, this can be interpreted by the fact that the parameters of a personalization model are usually estimated from sparse data; the resulting personalized prediction, despite of its low bias, is often volatile. In this paper, we study the problem further by investigating into the ranking of recommendation lists. From a risk management and portfolio retrieval perspective, there is no difference between the popularity-based and the personalized ranking as both of the recommendation outputs can be represented as the trade-off between expected relevance (reward) and associated uncertainty (risk). Through our analysis, we discover common scenarios and provide a technique to predict whether personalization will fail. Besides the theoretical understanding, our experimental results show that the resulting switch algorithm, which decides whether or not to personalize, outperforms the mainstream recommendation algorithms. Weinan Zhang 0001, Jun Wang 0012, Bowei Chen 0001 |
RecSys | 2 |
| 2013 | Optimizing top-n collaborative filtering via dynamic negative item samplingabstractCollaborative filtering techniques rely on aggregated user preference data to make personalized predictions. In many cases, users are reluctant to explicitly express their preferences and many recommender systems have to infer them from implicit user behaviors, such as clicking a link in a webpage or playing a music track. The clicks and the plays are good for indicating the items a user liked (i.e., positive training examples), but the items a user did not like (negative training examples) are not directly observed. Previous approaches either randomly pick negative training samples from unseen items or incorporate some heuristics into the learning model, leading to a biased solution and a prolonged training period. In this paper, we propose to dynamically choose negative training samples from the ranked list produced by the current prediction model and iteratively update our model. The experiments conducted on three large-scale datasets show that our approach not only reduces the training time, but also leads to significant performance gains. Weinan Zhang 0001, Tianqi Chen 0001, Jun Wang 0012, Yong Yu 0001 |
SIGIR | 3 |
| 2013 | Bias-variance decomposition of ir evaluationabstractIt has been recognized that, when an information retrieval (IR) system achieves improvement in mean retrieval effectiveness (e.g. mean average precision (MAP)) over all the queries, the performance (e.g., average precision (AP)) of some individual queries could be hurt, resulting in retrieval instability. Some stability/robustness metrics have been proposed. However, they are often defined separately from the mean effectiveness metric. Consequently, there is a lack of a unified formulation of effectiveness, stability and overall retrieval quality (considering both). In this paper, we present a unified formulation based on the bias-variance decomposition. Correspondingly, a novel evaluation methodology is developed to evaluate the effectiveness and stability in an integrated manner. A case study applying the proposed methodology to evaluation of query language modeling illustrates the usefulness and analytical power of our approach. Peng Zhang 0002, Dawei Song 0001, Jun Wang 0012, Yuexian Hou |
SIGIR | 3 |
| 2013 | Probabilistic group recommendation via information matchingabstractIncreasingly, web recommender systems face scenarios where they need to serve suggestions to groups of users; for example, when families share e-commerce or movie rental web accounts. Research to date in this domain has proposed two approaches: computing recommendations for the group by merging any members' ratings into a single profile, or computing ranked recommendations for each individual that are then merged via a range of heuristics. In doing so, none of the past approaches reason on the preferences that arise in individuals when they are members of a group. In this work, we present a probabilistic framework, based on the notion of information matching, for group recommendation. This model defines group relevance as a combination of the item's relevance to each user as an individual and as a member of the group; it can then seamlessly incorporate any group recommendation strategy in order to rank items for a set of individuals. We evaluate the model's efficacy at generating recommendations for both single individuals and groups using the MovieLens and MoviePilot data sets. In both cases, we compare our results with baselines and state-of-the-art collaborative filtering algorithms, and show that the model outperforms all others over a variety of ranking metrics. Jagadeesh Gorla, Neal Lathia, Stephen E. Robertson, Jun Wang 0012 |
WWW | 4 |
| 2013 | Interactive exploratory search for multi page search resultsabstractModern information retrieval interfaces typically involve multiple pages of search results, and users who are recall minded or engaging in exploratory search using ad hoc queries are likely to access more than one page. Document rankings for such queries can be improved by allowing additional context to the query to be provided by the user herself using explicit ratings or implicit actions such as clickthroughs. Existing methods using this information usually involved detrimental UI changes that can lower user satisfaction. Instead, we propose a new feedback scheme that makes use of existing UIs and does not alter user's browsing behaviour; to maximise retrieval performance over multiple result pages, we propose a novel retrieval optimisation framework and show that the optimal ranking policy should choose a diverse, exploratory ranking to display on the first page. Then, a personalised re-ranking of the next pages can be generated based on the user's feedback from the first page. We show that document correlations used in result diversification have a significant impact on relevance feedback and its effectiveness over a search session. TREC evaluations demonstrate that our optimal rank strategy (including approximative Monte Carlo Sampling) can naturally optimise the trade-off between exploration and exploitation and maximise the overall user's satisfaction over time against a number of similar baselines. Xiaoran Jin, Marc Sloan, Jun Wang 0012 |
WWW | 3 |
| 2013 | Bridging memory-based collaborative filtering and text retrieval
Alejandro Bellogín, Jun Wang 0012, Pablo Castells |
Inf. Retr. | 2 |
| 2012 | Sequential selection of correlated ads by POMDPsabstractOnline advertising has become a key source of revenue for both web search engines and online publishers. For them, the ability of allocating right ads to right webpages is critical because any mismatched ads would not only harm web users' satisfactions but also lower the ad income. In this paper, we study how online publishers could optimally select ads to maximize their ad incomes over time. The conventional offline, content-based matching between webpages and ads is a fine start but cannot solve the problem completely because good matching does not necessarily lead to good payoff. Moreover, with the limited display impressions, we need to balance the need of selecting ads to learn true ad payoffs (exploration) with that of allocating ads to generate high immediate payoffs based on the current belief (exploitation). In this paper, we address the problem by employing Partially observable Markov decision processes (POMDPs) and discuss how to utilize the correlation of ads to improve the efficiency of the exploration and increase ad incomes in a long run. Our mathematical derivation shows that the belief states of correlated ads can be naturally updated using a formula similar to collaborative filtering. To test our model, a real world ad dataset from a major search engine is collected and categorized. Experimenting over the data, we provide an analyse of the effect of the underlying parameters, and demonstrate that our algorithms significantly outperform other strong baselines. Shuai Yuan 0002, Jun Wang 0012 |
CIKM | 2 |
| 2012 | Top-k Retrieval Using Facility Location Analysis
Guido Zuccon, Leif Azzopardi, Dell Zhang, Jun Wang 0012 |
ECIR | 4 |
| 2012 | Adaptive diversification of recommendation results via latent factor portfolioabstractThis paper studies result diversification in collaborative filtering. We argue that the diversification level in a recommendation list should be adapted to the target users' individual situations and needs. Different users may have different ranges of interests -- the preference of a highly focused user might include only few topics, whereas that of the user with broad interests may encompass a wide range of topics. Thus, the recommended items should be diversified according to the interest range of the target user. Such an adaptation is also required due to the fact that the uncertainty of the estimated user preference model may vary significantly between users. To reduce the risk of the recommendation, we should take the difference of the uncertainty into account as well. Yue Shi 0002, Jun Wang 0012, Martha A. Larson, Alan Hanjalic |
SIGIR | 3 |
| 2012 | Precoding with Known Transmit Coupling and Spatial Covariance MatricesabstractA closed-form precoding scheme using the knowledge of transmit coupling and spatial covariance matrices is proposed, which ensures that the achievable rate increases with the number of transmit antennas when the size of the transmit antenna array is fixed and the radiated power, but not the amplifier output power, is constrained. Eigenvalue-proportional power allocation is employed in this precoding scheme. Simulation shows that the performance of eigenvalue-proportional power allocation is close to that of covariance matrix based waterfilling power allocation. Energy efficiency of different schemes is evaluated and compared. Fanglei Sun, Peng Shang, Jun Wang 0012 |
VTC Spring | 5 |
| 2012 | 2nd international workshop on diversity in document retrieval (DDR 2012)abstractWhen an ambiguous query is received, a sensible approach is for the information retrieval (IR) system to diversify the results retrieved for this query, in the hope that at least one of the interpretations of the query intent will satisfy the user. Diversity is an increasingly important topic, of interest to both academic researchers (such as participants in the TREC Web and Blog track diversity tasks, or the NTCIR INTENT task), as well as to search engines professionals. In the 2nd edition of the Diversity in Document Retrieval workshop (DDR 2012), we solicited submissions both on approaches and models for diversity, the evaluation of diverse search results, and on applications of diverse search results. This workshop builds upon a successful 1st edition of DDR which was held at ECIR 2011 in Dublin, Ireland. Craig Macdonald, Jun Wang 0012, Charles L. A. Clarke |
WSDM | 2 |
| 2012 | Using control theory for stable and efficient recommender systemsabstractThe aim of a web-based recommender system is to provide highly accurate and up-to-date recommendations to its users; in practice, it will hope to retain its users over time. However, this raises unique challenges. To achieve complex goals such as keeping the recommender model up-to-date over time, we need to consider a number of external requirements. Generally, these requirements arise from the physical nature of the system, for instance the available computational resources. Ideally, we would like to design a system that does not deviate from the required outcome. Modeling such a system over time requires to describe the internal dynamics as a combination of the underlying recommender model and the its users' behavior. We propose to solve this problem by applying the principles of modern control theory - a powerful set of tools to deal with dynamical systems - to construct and maintain a stable and robust recommender system for dynamically evolving environments. In particular, we introduce a design principle by focusing on the dynamic relationship between the recommender system's performance and the number of new training samples the system requires. This enables us to automate the control other external factors such as the system's update frequency. We show that, by using a Proportional-Integral-Derivative controller, a recommender system is able to automatically and accurately estimate the required input to keep the output close to a pre-defined requirements. Our experiments on a standard rating dataset show that, by using a feedback loop between system performance and training, the trade-off between the effectiveness and efficiency of the system can be well maintained. We close by discussing the widespread applicability of our approach to a variety of scenarios that recommender systems face. Tamas Jambor, Jun Wang 0012, Neal Lathia |
WWW | 2 |
| 2011 | Structured collaborative filteringabstractIn a general collaborative filtering (CF) setting, a user profile contains a set of previously rated items and is used to represent the user's interest. Unfortunately, most CF approaches ignore the underlying structure of user profiles. In this paper, we argue that a certain class of interest is best represented jointly by several items, drawing an analogy to "phrases" in text retrieval, which are not equivalent to the separate meaning of their words. At an alternative stance, we also consider the situation where, analogously to word synonyms, two items might be substitutable when representing a class of interest. We propose an approach integrating these two notions as opposing poles on a continuum spectrum. Upon this, we model the underlying structure in user profiles, drawing an analogy with text retrieval. The approach gives rise to a novel structured Vector Space Model for CF. We show that item-based CF approaches are a special case of the proposed method. Alejandro Bellogín, Jun Wang 0012, Pablo Castells |
CIKM | 2 |
| 2011 | Statistical information retrieval modelling: from the probability ranking principle to recent advances in diversity, portfolio theory, and beyondabstractStatistical modelling of Information Retrieval (IR) systems is a key driving force in the development of the IR field. The goal of this tutorial is to provide a comprehensive and up-to-date introduction to statistical IR modelling. We take a fresh and systematic perspective from the viewpoint of portfolio theory of IR and risk management. A unified treatment and new insights will be given to reflect the recent developments of considering the ranked retrieval results as a whole. Recent research progress in diversification, risk management, and portfolio theory will be covered, in addition to classic methods such as Maron and Kuhns' Probabilistic Indexing, Robertson-Sparck Jones model (and the resulting BM25 formula) and language modelling approaches. The tutorial also reviews the resulting practical algorithms of risk-aware query expansion, diverse ranking, IR metric optimization as well as their performance evaluations. Practical IR applications such as web search, multimedia retrieval, and collaborative filtering are also introduced, as well as discussion of new opportunities for future research and applications that intersect among information retrieval, knowledge management, and databases. Jun Wang 0012, Kevyn Collins-Thompson |
CIKM | 1 |
| 2011 | Text Retrieval Methods for Item Ranking in Collaborative Filtering
Alejandro Bellogín, Jun Wang 0012, Pablo Castells |
ECIR | 2 |
| 2011 | Workshop on novelty and diversity in recommender systems - DiveRS 2011abstractNovelty and diversity have been identified as key dimensions of recommendation utility in real scenarios, and a fundamental research direction to keep making progress in the field. Yet recommendation novelty and diversity remain a largely open area for research. The DiveRS workshop gathered researchers and practitioners interested in the role of these dimensions in recommender systems. The workshop seeks to advance towards a better understanding of what novelty and diversity are, how they can improve the effectiveness of recommendation methods and the utility of their outputs. The workshop pursued the identification of open problems, relevant research directions, and opportunities for innovation in the recommendation business. Pablo Castells, Jun Wang 0012, Rubén Lara, Dell Zhang |
RecSys | 2 |
| 2010 | Goal-Driven Collaborative Filtering - A Directional Error Based Approach
Tamas Jambor, Jun Wang 0012 |
ECIR | 2 |
| 2010 | Laplacian Co-hashing of Terms and Documents
Dell Zhang, Jun Wang 0012, Deng Cai 0001, Jinsong Lu |
ECIR | 2 |
| 2010 | Optimizing multiple objectives in collaborative filteringabstractThis paper is about the utility of making personalized recommendations. While it is important to accurately predict the target user's preference, in practice the accuracy should not be the only concern; a useful recommender system needs to consider the user's utility or satisfaction of fulfilling a certain information seeking task. For example, recommending popular items (products) is unlikely to result in more gain than discovering insignificant ("long tail") yet liked items because the popular ones might be already known to the user. Equally, recommending items that are out of stock would be frustrating for both the user and system if the system is employed to discover items to purchase. Thus, it is important to have a flexible recommendation framework that takes into account additional recommendation goals meanwhile minimizing the performance loss in order to provide greater adjustability and a better user experience. Tamas Jambor, Jun Wang 0012 |
RecSys | 2 |
| 2010 | On statistical analysis and optimization of information retrieval effectiveness metricsabstractThis paper presents a new way of thinking for IR metric optimization. It is argued that the optimal ranking problem should be factorized into two distinct yet interrelated stages: the relevance prediction stage and ranking decision stage. During retrieval the relevance of documents is not known a priori, and the joint probability of relevance is used to measure the uncertainty of documents' relevance in the collection as a whole. The resulting optimization objective function in the latter stage is, thus, the expected value of the IR metric with respect to this probability measure of relevance. Through statistically analyzing the expected values of IR metrics under such uncertainty, we discover and explain some interesting properties of IR metrics that have not been known before. Our analysis and optimization framework do not assume a particular (relevance) retrieval model and metric, making it applicable to many existing IR models and metrics. The experiments on one of resulting applications have demonstrated its significance in adapting to various IR metrics. Jun Wang 0012, Jianhan Zhu |
SIGIR | 1 |
| 2010 | Self-taught hashing for fast similarity searchabstractThe ability of fast similarity search at large scale is of great importance to many Information Retrieval (IR) applications. A promising way to accelerate similarity search is semantic hashing which designs compact binary codes for a large number of documents so that semantically similar documents are mapped to similar codes (within a short Hamming distance). Although some recently proposed techniques are able to generate high-quality codes for documents known in advance, obtaining the codes for previously unseen documents remains to be a very challenging problem. In this paper, we emphasise this issue and propose a novel Self-Taught Hashing (STH) approach to semantic hashing: we first find the optimal l-bit binary codes for all documents in the given corpus via unsupervised learning, and then train l classifiers via supervised learning to predict the l-bit code for any query document unseen before. Our experiments on three real-world text datasets show that the proposed approach using binarised Laplacian Eigenmap (LapEig) and linear Support Vector Machine (SVM) outperforms state-of-the-art techniques significantly. Dell Zhang, Jun Wang 0012, Deng Cai 0001, Jinsong Lu |
SIGIR | 2 |
| 2010 | Personalization of tagging systems
Jun Wang 0012, Maarten Clements, Jie Yang 0015, Arjen P. de Vries, Marcel J. T. Reinders |
Inf. Process. Manag. | 1 |
| 2009 | Bridging the gap: complex networks meet information and knowledge managementabstractIn this article, we briefly summarize the motivation, content and structure of the CNIKM'09 workshop. Jun Wang 0012, Shi Zhou, Dell Zhang |
CIKM | 1 |
| 2009 | Mean-Variance Analysis: A New Document Ranking Theory in Information Retrieval
Jun Wang 0012 |
ECIR | 1 |
| 2009 | Risk-Aware Information Retrieval
Jianhan Zhu, Jun Wang 0012, Michael J. Taylor 0001, Ingemar J. Cox |
ECIR | 2 |
| 2009 | Portfolio theory of information retrievalabstractThis paper studies document ranking under uncertainty. It is tackled in a general situation where the relevance predictions of individual documents have uncertainty, and are dependent between each other. Inspired by the Modern Portfolio Theory, an economic theory dealing with investment in financial markets, we argue that ranking under uncertainty is not just about picking individual relevant documents, but about choosing the right combination of relevant documents. This motivates us to quantify a ranked list of documents on the basis of its expected overall relevance (mean) and its variance; the latter serves as a measure of risk, which was rarely studied for document ranking in the past. Through the analysis of the mean and variance, we show that an optimal rank order is the one that balancing the overall relevance (mean) of the ranked list against its risk level (variance). Based on this principle, we then derive an efficient document ranking algorithm. It generalizes the well-known probability ranking principle (PRP) by considering both the uncertainty of relevance predictions and correlations between retrieved documents. Moreover, the benefit of diversification is mathematically quantified; we show that diversifying documents is an effective way to reduce the risk of document ranking. Experimental results in text retrieval confirm performance. Jun Wang 0012, Jianhan Zhu |
SIGIR | 1 |
| 2009 | Risky business: modeling and exploiting uncertainty in information retrievalabstractMost retrieval models estimate the relevance of each document to a query and rank the documents accordingly. However, such an approach ignores the uncertainty associated with the estimates of relevancy. If a high estimate of relevancy also has a high uncertainty, then the document may be very relevant or not relevant at all. Another document may have a slightly lower estimate of relevancy but the corresponding uncertainty may be much less. In such a circumstance, should the retrieval engine risk ranking the first document highest, or should it choose a more conservative (safer) strategy that gives preference to the second document? There is no definitive answer to this question, as it depends on the risk preferences of the user and the information retrieval system. In this paper we present a general framework for modeling uncertainty and introduce an asymmetric loss function with a single parameter that can model the level of risk the system is willing to accept. By adjusting the risk preference parameter, our approach can effectively adapt to users' different retrieval strategies. Jianhan Zhu, Jun Wang 0012, Ingemar J. Cox, Michael J. Taylor 0001 |
SIGIR | 2 |
| 2009 | Topic (query) selection for IR evaluationabstractThe need for evaluating large amounts of topics (queries) makes IR evaluation an uneasy task. In this paper, we study a topic selection problem for IR evaluation. The selection criterion is based on the overall difficulty of the chosen set, as well as the uncertainty of the final IR metric applied to the systems. Our preliminary experiments demonstrate that our approach helps to identify a set of topics that provides confident estimates of systems' performance while keeping the requirement of the query difficulty. Jianhan Zhu, Jun Wang 0012, Vishwa Vinay, Ingemar J. Cox |
SIGIR | 2 |
| 2008 | Generation of Query-Biased Concepts Using Content and Structure for Query Reformulation
Youjin Chang, Jun Wang 0012, Mounia Lalmas-Roelleke |
NLDB | 2 |
| 2008 | TRIBLER: a social-based peer-to-peer systemabstractAbstract Most current peer‐to‐peer (P2P) file‐sharing systems treat their users as anonymous, unrelated entities, and completely disregard any social relationships between them. However, social phenomena such as friendship and the existence of communities of users with similar tastes or interests may well be exploited in such systems in order to increase their usability and performance. In this paper we present a novel social‐based P2P file‐sharing paradigm that exploits social phenomena by maintaining social networks and using these in content discovery, content recommendation, and downloading. Based on this paradigm's main concepts such as taste buddies and friends, we have designed and implemented the TRIBLER P2P file‐sharing system as a set of extensions to BitTorrent. We present and discuss the design of TRIBLER, and we show evidence that TRIBLER enables fast content discovery and recommendation at a low additional overhead, and a significant improvement in download performance. Copyright © 2007 John Wiley & Sons, Ltd. Johan A. Pouwelse, Pawel Garbacki, Jun Wang 0012, Arno Bakker, Jie Yang 0015, Alexandru Iosup, Dick H. J. Epema, Marcel J. T. Reinders, Maarten van Steen, Henk J. Sips |
Concurr. Comput. Pract. Exp. | 3 |
| 2008 | Probabilistic relevance ranking for collaborative filtering
Jun Wang 0012, Stephen E. Robertson, Arjen P. de Vries, Marcel J. T. Reinders |
Inf. Retr. | 1 |
| 2008 | Personalization on a peer-to-peer television system
Jun Wang 0012, Johan A. Pouwelse, Jenneke Fokker, Arjen P. de Vries, Marcel J. T. Reinders |
Multim. Tools Appl. | 1 |
| 2008 | Unified relevance models for rating prediction in collaborative filteringabstractCollaborative filtering aims at predicting a user's interest for a given item based on a collection of user profiles. This article views collaborative filtering as a problem highly related to information retrieval, drawing an analogy between the concepts of users and items in recommender systems and queries and documents in text retrieval. We present a probabilistic user-to-item relevance framework that introduces the concept of relevance into the related problem of collaborative filtering. Three different models are derived, namely, auser-based, anitem-based, and aunified relevance model, and we estimate their rating predictions from three sources: the user's own ratings for different items, other users' ratings for the same item, and ratings from different but similar users for other but similar items. To reduce the data sparsity encountered when estimating the probability density function of the relevance variable, we apply the nonparametric (data-driven) density estimation technique known as theParzen-window method(or kernel-based density estimation). Using a Gaussian window function, the similarity between users and/or items would, however, be based on Euclidean distance. Because the collaborative filtering literature has reported improved prediction accuracy when using cosine similarity, we generalize the Parzen-window method by introducing aprojection kernel. Existing user-based and item-based approaches correspond to two simplified instantiations of our framework. User-based and item-based collaborative filterings represent only a partial view of the prediction problem, where the unified relevance model brings these partial views together under the same umbrella. Experimental results complement the theoretical insights with improved recommendation accuracy. The unified model is more robust to data sparsity because the different types of ratings are used in concert. Jun Wang 0012, Arjen P. de Vries, Marcel J. T. Reinders |
ACM Trans. Inf. Syst. | 1 |
| 2006 | A User-Item Relevance Model for Log-Based Collaborative Filtering
Jun Wang 0012, Arjen P. de Vries, Marcel J. T. Reinders |
ECIR | 1 |
| 2006 | Unifying user-based and item-based collaborative filtering approaches by similarity fusionabstractMemory-based methods for collaborative filtering predict new ratings by averaging (weighted) ratings between, respectively, pairs of similar users or items. In practice, a large number of ratings from similar users or similar items are not available, due to the sparsity inherent to rating data. Consequently, prediction quality can be poor. This paper re-formulates the memory-based collaborative filtering problem in a generative probabilistic framework, treating individual user-item ratings as predictors of missing ratings. The final rating is estimated by fusing predictions from three sources: predictions based on ratings of the same item by other users, predictions based on different item ratings made by the same user, and, third, ratings predicted based on data from other but similar users rating other but similar items. Existing user-based and item-based approaches correspond to the two simple cases of our framework. The complete model is however more robust to data sparsity, because the different types of ratings are used in concert, while additional ratings from similar users towards similar items are employed as a background model to smooth the predictions. Experiments demonstrate that the proposed methods are indeed more robust against data sparsity and give better recommendations. Jun Wang 0012, Arjen P. de Vries, Marcel J. T. Reinders |
SIGIR | 1 |
| 2006 | Experiential Sampling in Multimedia SystemsabstractMultimedia systems must deal with multiple data streams. Each data stream usually contains significant volume of redundant noisy data. In many real-time applications, it is essential to focus the computing resources on a relevant subset of data streams at any given time instant and use it to build the model of the environment. We formulate this problem as an experiential sampling problem and propose an approach to utilize computing resources efficiently on the most informative subset of data streams. First, in this paper, we focus on theoretical background and develop a theoretical framework for a single data stream. We generalize the notion of static visual attention in a dynamical systems setting and propose a dynamical attention-orientated analysis method. This is achieved by a sampling representation that utilizes the current context and past experience for attention evolution. Hence, the multimedia analysis task at hand can select its data of interest while immediately discarding the irrelevant data to achieve efficiency and adaptability. Mohan Kankanhalli, Jun Wang 0012, Ramesh Jain 0001 |
IEEE Trans. Multim. | 2 |
| 2006 | Experiential Sampling on Multiple Data StreamsabstractMultimedia systems must deal with multiple data streams. Each data stream usually contains significant volume of redundant noisy data. In many real-time applications, it is essential to focus the computing resources on a relevant subset of data streams at any given time instant and use it to build the model of the environment. We formulate this problem as an experiential sampling problem and propose an approach to utilize computing resources efficiently on the most informative subset of data streams. In this paper, we generalize our experiential sampling framework to multiple data streams and provide an evaluation measure for this technique. We have successfully applied this framework to the problems of traffic monitoring, face detection and monologue detection. Mohan Kankanhalli, Jun Wang 0012, Ramesh Jain 0001 |
IEEE Trans. Multim. | 2 |
| 2005 | Self-organizing distributed collaborative filteringabstractWe propose a fully decentralized collaborative filtering approach that is self-organizing and operates in a distributed way. The relevances between downloading files (items) are stored locally at these items in so called item-based buddy tables and are updated each time that the items are downloaded. We then propose to use the language model to build recommendations for the different users based on the buddy tables of those items a user has downloaded previously. We have tested and compared our distributed collaborative filtering approach to centralized collaborative filtering and showed that it has similar performance. It is therefore a promising technique to facilitate recommendations in peer-to-peer networks. Jun Wang 0012, Marcel J. T. Reinders, Reginald L. Lagendijk, Johan A. Pouwelse |
SIGIR | 1 |
| 2005 | Analogies based video editing
Wei Qi Yan 0001, Mohan Kankanhalli, Jun Wang 0012 |
Multim. Syst. | 3 |
| 2005 | Automatic video logo detection and removal
Wei Qi Yan 0001, Jun Wang 0012, Mohan Kankanhalli |
Multim. Syst. | 2 |
| 2004 | Video content representation on tiny devicesabstractThe perceptual satisfaction of a user watching video on a tiny mobile device is constrained by the display capability and network bandwidth. To maximize the user's perceptual satisfaction in this constrained environment, we propose a new method to represent the video content adaptively in real-time on tiny devices according to the user's attention. First, a sampling based dynamic attention model is proposed to obtain and maintain the user's attention in the video streams. Second, based on the most attended regions and sequences extracted, the attention based representation is introduced to achieve a higher perceptual satisfaction on a small display. Experiments with users show the effectiveness of our proposed method in a video surveillance application. Jun Wang 0012, Marcel J. T. Reinders, Reginald L. Lagendijk, Jasper Lindenberg, Mohan Kankanhalli |
ICME | 1 |
| 2003 | A hierarchical framework for face tracking using state vector fusion for compressed videoabstractFaces usually are the most interesting objects in certain categories of video, like home videos and news clips. A novel sensor fusion based face tracking system is presented that tracks faces in compressed video, and aids automatic video indexing. Tracking is done by fusing the measurements from three independent sensors - motion and colour based trackers (Achanta, R. et al., IEEE Int. Conf. on Multimedia and Expo, 2002) and a face detector (Wang, J. et al., Proc. Int. Workshop on Advanced Image Technology, 2002) using a novel hierarchical framework based on Kalman filter state vector fusion. The tracking results show that the fused results are better than those of any individual sensors or their mean. Jun Wang 0012, Radhakrishna S. V. Achanta, Mohan Kankanhalli, Philippe Mulhem |
ICASSP (3) | 1 |
| 2003 | Experience based sampling technique for multimedia analysisabstractWe present a novel experience based sampling or experiential sampling technique which has the ability to focus on the analysis's task by making use of the contextual information from the environment. In this technique, sensor samples are used to gather information about the current environment and attention samples are used to represent the current state of attention. The task-attended samples are inferred from experience and maintained by a sampling based dynamical system. The multimedia analysis task can then focus on the attention samples only. Moreover, past experiences and the current environment can be used to adaptively correct and tune the attention. Experimental results have been presented to demonstrate the efficacy of our technique. Jun Wang 0012, Mohan Kankanhalli |
ACM Multimedia | 1 |