49 citations · 116 across the 26 of their papers we have counts for
26 papers
APeB: Benchmarking Personalization Ability of Large Language Model Agents
Garry Yang, Zizhe Chen, Xinru Chen +9
LLM-powered agents struggle with personalization when users issue raw, underspecified queries. In this setting, agents must infer latent intent, extract preferences from noisy inte…
On Information Self-Locking in Reinforcement Learning for Active Reasoning of LLM agents
Deyu Zou, Yongqiang Chen, Fan Feng +4
Reinforcement learning (RL) has become a de facto paradigm for building LLM-based agents that act, interact, and reason over extended task horizons. However, in active reasoning wh…
When and Why Does Multi-Agent Debate Fail and Does It Really Underperform?
Yongqiang Chen, Gang Niu, James Cheng +2
Multi-agent debate (MAD) was proposed as a promising approach for ensembling the wisdom of multiple large language models (LLMs) to improve reasoning and provide effective supervis…
Reducing Belief Deviation in Reinforcement Learning for Active Reasoning
Deyu Zou, Yongqiang Chen, Jianxiang Wang +5
Active reasoning requires large language model (LLM) agents to interact with external sources and strategically gather information to solve problems in multiple turns. Central to t…
Weak-to-Strong GraphRAG: Aligning Weak Retrievers with Large Language Models for Graph-based Retrieval Augmented Generation
Deyu Zou, Yongqiang Chen, Mufei Li +5
Graph-based retrieval-augmented generation (RAG) enables large language models (LLMs) to ground responses with structured external knowledge from up-to-date knowledge graphs (KGs)…
On the Thinking-Language Modeling Gap in Large Language Models
Chenxi Liu, Yongqiang Chen, Tongliang Liu +3
System 2 reasoning is one of the defining characteristics of intelligence, which requires slow and logical thinking. Human conducts System 2 reasoning via the language of thoughts…
Retrieval-Augmented Generation with Hierarchical Knowledge
Haoyu Huang, Yongfeng Huang, Junjie Yang +5
Graph-based Retrieval-Augmented Generation (RAG) methods have significantly enhanced the performance of large language models (LLMs) in domain-specific tasks. However, existing RAG…
BrainOOD: Out-of-distribution Generalizable Brain Network Analysis
Jiaxing Xu, Yongqiang Chen, Xia Dong +5
In neuroscience, identifying distinct patterns linked to neurological disorders, such as Alzheimer's and Autism, is critical for early diagnosis and effective intervention. Graph N…
DivIL: Unveiling and Addressing Over-Invariance for Out-of- Distribution Generalization
Jiaqi Wang, Yuhang Zhou, Zhixiong Zhang +3
Out-of-distribution generalization is a common problem that expects the model to perform well in the different distributions even far from the train data. A popular approach to add…
Eliciting Causal Abilities in Large Language Models for Reasoning Tasks
Yajing Wang, Zongwei Luo, Jingzhe Wang +3
Prompt optimization automatically refines prompting expressions, unlocking the full potential of LLMs in downstream tasks. However, current prompt optimization methods are costly t…
UniMoT: Unified Molecule-Text Language Model with Discrete Token Representation
Shuhan Guo, Yatao Bian, Ruibing Wang +3
The remarkable success of Large Language Models (LLMs) across diverse tasks has driven the research community to extend their capabilities to molecular applications. However, most…
How Interpretable Are Interpretable Graph Neural Networks?
Yongqiang Chen, Yatao Bian, Bo Han +1
Interpretable graph neural networks (XGNNs ) are widely adopted in various scientific applications involving graph-structured data. Existing XGNNs predominantly adopt the attention…
HIGHT: Hierarchical Graph Tokenization for Molecule-Language Alignment
Yongqiang Chen, Quanming Yao, Juzheng Zhang +2
Recently, there has been a surge of interest in extending the success of large language models (LLMs) from texts to molecules. Most existing approaches adopt a graph neural network…
A Sober Look at the Robustness of CLIPs to Spurious Features
Qizhou Wang, Yong Lin, Yongqiang Chen +3
Large vision language models, such as CLIP, demonstrate impressive robustness to spurious features than single-modal models trained on ImageNet. However, existing test datasets are…
Discovering and Reasoning of Causality in the Hidden World with Large Language Models
Chenxi Liu, Yongqiang Chen, Tongliang Liu +4
Revealing hidden causal variables alongside the underlying causal mechanisms is essential to the development of science. Despite the progress in the past decades, existing practice…
Enhancing Neural Subset Selection: Integrating Background Information into Set Representations
Binghui Xie, Yatao Bian, Kaiwen zhou +5
Learning neural subset selection tasks, such as compound selection in AI-aided drug discovery, have become increasingly pivotal across diverse applications. The existing methodolog…
Enhancing Evolving Domain Generalization through Dynamic Latent Representations
Binghui Xie, Yongqiang Chen, Jiaqi Wang +4
Domain generalization is a critical challenge for machine learning systems. Prior domain generalization methods focus on extracting domain-invariant features across several station…
Positional Information Matters for Invariant In-Context Learning: A Case Study of Simple Function Classes
Yongqiang Chen, Binghui Xie, Kaiwen Zhou +3
In-context learning (ICL) refers to the ability of a model to condition on a few in-context demonstrations (input-output examples of the underlying task) to generate the answer for…
Does Invariant Graph Learning via Environment Augmentation Learn Invariance?
Yongqiang Chen, Yatao Bian, Kaiwen Zhou +3
Invariant graph representation learning aims to learn the invariance among data from different environments for out-of-distribution generalization on graphs. As the graph environme…
Towards out-of-distribution generalizable predictions of chemical kinetics properties
Zihao Wang, Yongqiang Chen, Yang Duan +4
Machine Learning (ML) techniques have found applications in estimating chemical kinetic properties. With the accumulated drug molecules identified through "AI4drug discovery", the…
Understanding and Improving Feature Learning for Out-of-Distribution Generalization
Yongqiang Chen, Wei Huang, Kaiwen Zhou +3
A common explanation for the failure of out-of-distribution (OOD) generalization is that the model trained with empirical risk minimization (ERM) learns spurious features instead o…
Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization
Yongqiang Chen, Kaiwen Zhou, Yatao Bian +8
Recently, there has been a growing surge of interest in enabling machine learning systems to generalize well to Out-of-Distribution (OOD) data. Most efforts are devoted to advancin…
Understanding and Improving Graph Injection Attack by Promoting Unnoticeability
Yongqiang Chen, Han Yang, Yonggang Zhang +4
Recently Graph Injection Attack (GIA) emerges as a practical attack scenario on Graph Neural Networks (GNNs), where the adversary can merely inject few malicious nodes instead of m…
Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphs
Yongqiang Chen, Yonggang Zhang, Yatao Bian +6
Despite recent success in using the invariance principle for out-of-distribution (OOD) generalization on Euclidean data (e.g., images), studies on graph data are still limited. Dif…