← Search

Jundong Li

55 accepted papers

2026

ConvRec-R1: Training LLM-based Conversational Recommender Systems with Reinforcement Learning

ICLR 2026poster

Large language models (LLMs) are reshaping the recommender system paradigm by enabling users to express preferences and receive recommendations through conversations. Yet, aligning LLMs to the recommendation task remains challenging: pretrained LLMs often generate out-of-catalog items, violate requi…

Cited by 0SourcecodeScholar
2026

Energy-Based Transformers are Scalable Learners and Thinkers

ICLR 2026oral

Inference-time computation, analogous to human System 2 Thinking, has recently become popular for improving model performance. However, most existing approaches suffer from several limitations: they are modality-specific (e.g., working only in text), problem-specific (e.g., verifiable domains like m…

Cited by 0SourcecodeScholar
2026

IAPO: Information-Aware Policy Optimization for Token-Efficient Reasoning

ICML 2026poster

Large language models increasingly rely on long chains of thought to improve accuracy, yet such gains come with substantial inference-time costs. We revisit token-efficient post-training and argue that existing sequence-level reward-shaping methods offer limited control over how reasoning effort is …

Cited by 0SourceScholar
2026

Probing to Refine: Reinforcement Distillation of LLM Reasoners via Explanatory Inversion

ICLR 2026poster

Distilling robust reasoning capabilities from large language models (LLMs) into smaller, computationally efficient student models remains an unresolved challenge. Despite recent advances, distilled models frequently suffer from superficial pattern memorization and subpar generalization. To overcome…

Cited by 0SourceScholar
2026

Reforming the Mechanism: Editing Reasoning Patterns in LLMs with Circuit Reshaping

ICLR 2026poster

Large language models (LLMs) often exhibit flawed reasoning ability that undermines reliability. Existing approaches to improving reasoning typically treat it as a general and monolithic skill, applying broad training that is inefficient and unable to target specific reasoning errors. We introduce R…

Cited by 0SourcecodeScholar
2026

SMILE: Extended Deep Submodular Function-Based Instruction and In-context Learning Demonstration Selection

ICML 2026poster

Prompt optimization is a key way to steer large language models when fine-tuning is impractical. However, instruction optimization (IO) and in-context learning (ICL) demonstration selection are often optimized separately and combined post hoc, implicitly assuming that a "best'' instruction and a "be…

Cited by 0SourceScholar
2025

AnyMAC: Cascading Flexible Multi-Agent Collaboration via Next-Agent Prediction

EMNLP 2025

Recent progress in large language model (LLM)-based multi-agent collaboration highlights the power of structured communication in enabling collective intelligence. However, existing methods largely rely on static or graph-based inter-agent topologies, lacking the potential adaptability and flexibili

2025

Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusion

ICML 2025spotlight

Symmetry in the parameter space of deep neural networks (DNNs) has proven beneficial for various deep learning applications. A well-known example is the permutation symmetry in Multi-Layer Perceptrons (MLPs), where permuting the rows of weight matrices in one layer and applying the inverse permutati…

2025

BrainMAP: Learning Multiple Activation Pathways in Brain Networks

AAAI 2025technical

Functional Magnetic Resonance Image (fMRI) is commonly employed to study human brain activity, since it offers insight into the relationship between functional fluctuations and human behavior. To enhance analysis and comprehension of brain activity, Graph Neural Networks (GNNs) have been widely appl…

2025

CEB: Compositional Evaluation Benchmark for Fairness in Large Language Models

ICLR 2025spotlight

As Large Language Models (LLMs) are increasingly deployed to handle various natural language processing (NLP) tasks, concerns regarding the potential negative societal impacts of LLM-generated content have also arisen. To evaluate the biases exhibited by LLMs, researchers have recently proposed a va…

Cited by 12SourcePDFScholar
2025

Causal Effect Estimation with Mixed Latent Confounders and Post-treatment Variables

ICLR 2025poster

Causal inference from observational data has attracted considerable attention among researchers. One main obstacle is the handling of confounders. As direct measurement of confounders may not be feasible, recent methods seek to address the confounding bias via proxy variables, i.e., covariates postu…

Cited by 0SourcePDFScholar
2025

CoRAG: Enhancing Hybrid Retrieval-Augmented Generation through a Cooperative Retriever Architecture

EMNLP 2025

Retrieval-Augmented Generation (RAG) is introduced to enhance Large Language Models (LLMs) by integrating external knowledge. However, conventional RAG approaches treat retrieved documents as independent units, often overlooking their interdependencies. Hybrid-RAG, a recently proposed paradigm that

2025

From Cross-Task Examples to In-Task Prompts: A Graph-Based Pseudo-Labeling Framework for In-context Learning

EMNLP 2025

The capability of in-context learning (ICL) enables large language models (LLMs) to perform novel tasks without parameter updates by conditioning on a few input-output examples. However, collecting high-quality examples for new or challenging tasks can be costly and labor-intensive. In this work, we

2025

Graph Neural Networks Are More Than Filters: Revisiting and Benchmarking from A Spectral Perspective

ICLR 2025poster

Graph Neural Networks (GNNs) have achieved remarkable success in various graph-based learning tasks. While their performance is often attributed to the powerful neighborhood aggregation mechanism, recent studies suggest that other components such as non-linear layers may also significantly affecting…

2025

GraphTOP: Graph Topology-Oriented Prompting for Graph Neural Networks

NeurIPS 2025poster

Graph Neural Networks (GNNs) have revolutionized the field of graph learning by learning expressive graph representations from massive graph data. As a common pattern to train powerful GNNs, the "pre-training, adaptation" scheme first pre-trains GNNs over unlabeled graph data and subsequently adapts…

Cited by 0SourcecodeScholar
2025

Harnessing Large Language Models for Disaster Management: A Survey

ACL 2025finding

Large Language Models (LLMs) have demonstrated remarkable capabilities across various domains, including their emerging role in mitigating threats to human life, infrastructure, and the environment during natural disasters. Despite increasing research on disaster-focused LLMs, there remains a lack o…

Cited by 0SourcePDFScholar
2025

Hierarchical Demonstration Order Optimization for Many-shot In-Context Learning

NeurIPS 2025poster

In-Context Learning (ICL) is a technique where large language models (LLMs) leverage multiple demonstrations (i.e., examples) to perform tasks. With the recent expansion of LLM context windows, many-shot ICL (generally with more than 50 demonstrations) can lead to significant performance improvement…

Cited by 0SourcecodeScholar
2025

Interpreting Pretrained Language Models via Concept Bottlenecks (Extended Abstract)

IJCAI 2025

Pretrained language models (PLMs) achieve state-of-the-art results but often function as ``black boxes'', hindering interpretability and responsible deployment. While methods like attention analysis exist, they often lack clarity and intuitiveness. We propose interpreting PLMs through high-level, hu

2025

LLM-based Conversational Recommendation Agents with Collaborative Verbalized Experience

EMNLP 2025

Large language models (LLMs) have demonstrated impressive zero-shot capabilities in conversational recommender systems (CRS). However, effectively utilizing historical conversations remains a significant challenge. Current approaches either retrieve few-shot examples or extract global rules to enhan

2025

Learning from Diverse Reasoning Paths with Routing and Collaboration

EMNLP 2025

Advances in large language models (LLMs) significantly enhance reasoning capabilities but their deployment is restricted in resource-constrained scenarios. Knowledge distillation addresses this by transferring knowledge from powerful teacher models to compact and transparent students.However, effect

2025

MAPLE: Many-Shot Adaptive Pseudo-Labeling for In-Context Learning

ICML 2025poster

In-Context Learning (ICL) empowers Large Language Models (LLMs) to tackle diverse tasks by incorporating multiple input-output examples, known as demonstrations, into the input of LLMs. More recently, advancements in the expanded context windows of LLMs have led to many-shot ICL, which uses hundreds…

2025

Question-Aware Knowledge Graph Prompting for Enhancing Large Language Models

ACL 2025finding

Large Language Models (LLMs) often struggle with tasks requiring external knowledge, such as knowledge-intensive Multiple Choice Question Answering (MCQA). Integrating Knowledge Graphs (KGs) can enhance reasoning; however, existing methods typically demand costly fine-tuning or retrieve noisy KG inf…

2025

Reasoning of Large Language Models over Knowledge Graphs with Super-Relations

ICLR 2025poster

While large language models (LLMs) have made significant progress in processing and reasoning over knowledge graphs, current methods suffer from a high non-retrieval rate. This limitation reduces the accuracy of answering questions based on these graphs. Our analysis reveals that the combination of…

2025

ST-FiT: Inductive Spatial-Temporal Forecasting with Limited Training Data

AAAI 2025technical

Spatial-temporal graphs are widely used in a variety of real-world applications. Spatial-Temporal Graph Neural Networks (STGNNs) have emerged as a powerful tool to extract meaningful insights from this data. However, in real-world applications, most nodes may not possess any available temporal data…

2025

SemCoT: Accelerating Chain-of-Thought Reasoning through Semantically-Aligned Implicit Tokens

NeurIPS 2025poster

Chain-of-Thought (CoT) enhances the performance of Large Language Models (LLMs) on reasoning tasks by encouraging step-by-step solutions. However, the verbosity of CoT reasoning hinders its mass deployment in efficiency-critical applications. Recently, implicit CoT approaches have emerged, which enc…

Cited by 0SourcecodeScholar
2025

Separate the Wheat from the Chaff: Winnowing Down Divergent Views in Retrieval Augmented Generation

EMNLP 2025

Retrieval-augmented generation (RAG) addresses the limitation of large language models (LLMs) in achieving up-to-date information by integrating external knowledge sources, but it is hindered by noisy or irrelevant retrieved data, leading to reduced accuracy. Additionally, most RAG methods rely on t

Cited by 0SourcePDFScholar
2025

Towards Global-level Mechanistic Interpretability: A Perspective of Modular Circuits of Large Language Models

ICML 2025poster

Mechanistic interpretability (MI) research aims to understand large language models (LLMs) by identifying computational circuits, subgraphs of model components with associated functional interpretations, that explain specific behaviors. Current MI approaches focus on discovering task-specific circui…

Cited by 0SourcePDFScholar
2025

Virtual Nodes Can Help: Tackling Distribution Shifts in Federated Graph Learning

AAAI 2025technical

Federated Graph Learning (FGL) enables multiple clients to jointly train powerful graph learning models, e.g., Graph Neural Networks (GNNs), without sharing their local graph data for graph-related downstream tasks, such as graph property prediction. In the real world, however, the graph data can su…

2024

Adversarial Attacks on Fairness of Graph Neural Networks

ICLR 2024poster

Fairness-aware graph neural networks (GNNs) have gained a surge of attention as they can reduce the bias of predictions on any demographic group (e.g., female) in graph-based applications. Although these methods greatly improve the algorithmic fairness of GNNs, the fairness can be easily corrupted b…

2024

Efficient Prompt Optimization Through the Lens of Best Arm Identification

NeurIPS 2024poster

The remarkable instruction-following capability of large language models (LLMs) has sparked a growing interest in automatically finding good prompts, i.e., prompt optimization. Most existing works follow the scheme of selecting from a pre-generated pool of candidate prompts. However, these designs m…

Cited by 7SourcePDFScholar
2024

Explaining Graph Neural Networks with Large Language Models: A Counterfactual Perspective on Molecule Graphs

EMNLP 2024finding

In recent years, Graph Neural Networks (GNNs) have become successful in molecular property prediction tasks such as toxicity analysis. However, due to the black-box nature of GNNs, their outputs can be concerning in high-stakes decision-making scenarios, e.g., drug discovery. Facing such an issue, G…

2024

FastGAS: Fast Graph-based Annotation Selection for In-Context Learning

ACL 2024findings

In-context learning (ICL) empowers large language models (LLMs) to tackle new tasks by using a series of training instances as prompts. Since generating the prompts needs to sample from a vast pool of instances and annotate them (e.g., add labels in classification task), existing methods have propos…

Cited by 3SourcePDFScholar
2024

Few-shot Knowledge Graph Relational Reasoning via Subgraph Adaptation

NAACL 2024long

Few-shot Knowledge Graph (KG) Relational Reasoning aims to predict unseen triplets (i.e., query triplets) for rare relations in KGs, given only several triplets of these relations as references (i.e., support triplets). This task has gained significant traction due to the widespread use of knowledge…

2024

Glue pizza and eat rocks - Exploiting Vulnerabilities in Retrieval-Augmented Generative Models

EMNLP 2024main

Retrieval-Augmented Generative (RAG) models enhance Large Language Models (LLMs) by integrating external knowledge bases, improving their performance in applications like fact-checking and information searching. In this paper, we demonstrate a security threat where adversaries can exploit the openne…

Cited by 4SourcePDFScholar
2024

Knowledge Graph-Enhanced Large Language Models via Path Selection

ACL 2024findings

Large Language Models (LLMs) have shown unprecedented performance in various real-world applications. However, they are known to generate factually inaccurate outputs, a.k.a. the hallucination problem. In recent years, incorporating external knowledge extracted from Knowledge Graphs (KGs) has become…

2024

Large Language Models for Data Annotation and Synthesis: A Survey

EMNLP 2024main

Data annotation and synthesis generally refers to the labeling or generating of raw data with relevant information, which could be used for improving the efficacy of machine learning models. The process, however, is labor-intensive and costly. The emergence of advanced Large Language Models (LLMs),…

2024

Mixture of Demonstrations for In-Context Learning

NeurIPS 2024poster

In-Context Learning (ICL) empowers Large Language Models (LLMs) to tackle various tasks by providing input-output examples as additional inputs, referred to as demonstrations. Nevertheless, the performance of ICL could be easily impacted by the quality of selected demonstrations. Existing efforts ge…

Cited by 4SourcePDFScholar
2024

Towards Certified Unlearning for Deep Neural Networks

ICML 2024poster

In the field of machine unlearning, certified unlearning has been extensively studied in convex machine learning models due to its high efficiency and strong theoretical guarantees. However, its application to deep neural networks (DNNs), known for their highly nonconvex nature, still poses challeng…

2023

BIC: Twitter Bot Detection with Text-Graph Interaction and Semantic Consistency

ACL 2023long

Twitter bots are automatic programs operated by malicious actors to manipulate public opinion and spread misinformation. Research efforts have been made to automatically identify bots based on texts and networks on social media. Existing methods only leverage texts or networks alone, and while few w…

2023

Interpreting Unfairness in Graph Neural Networks via Training Node Attribution

AAAI 2023technical

Graph Neural Networks (GNNs) have emerged as the leading paradigm for solving graph analytical problems in various real-world applications. Nevertheless, GNNs could potentially render biased predictions towards certain demographic subgroups. Understanding how the bias in predictions arises is criti…

2023

Learning Causal Effects on Hypergraphs (Extended Abstract)

IJCAI 2023poster

Hypergraphs provide an effective abstraction for modeling multi-way group interactions among nodes, where each hyperedge can connect any number of nodes. Different from most existing studies which leverage statistical dependencies, we study hypergraphs from the perspective of causality. Specifically…

Cited by 0SourcePDFScholar
2023

Noise-Robust Fine-Tuning of Pretrained Language Models via External Guidance

EMNLP 2023long findings

Adopting a two-stage paradigm of pretraining followed by fine-tuning, Pretrained Language Models (PLMs) have achieved substantial advancements in the field of natural language processing. However, in real-world scenarios, data labels are often noisy due to the complex annotation process, making it e…

Cited by 0SourceScholar
2022

BOND: Benchmarking Unsupervised Outlier Node Detection on Static Attributed Graphs

NeurIPS 2022accept

Detecting which nodes in graphs are outliers is a relatively new machine learning task with numerous applications. Despite the proliferation of algorithms developed in recent years for this task, there has been no standard comprehensive setting for performance evaluation. Consequently, it has been d…

2022

CLEAR: Generative Counterfactual Explanations on Graphs

NeurIPS 2022accept

Counterfactual explanations promote explainability in machine learning models by answering the question “how should the input instance be altered to obtain a desired predicted label?". The comparison of this instance before and after perturbation can enhance human interpretation. Most existing studi…

Cited by 73SourcePDFScholar
2022

FAITH: Few-Shot Graph Classification with Hierarchical Task Graphs

IJCAI 2022poster

Few-shot graph classification aims at predicting classes for graphs, given limited labeled graphs for each class. To tackle the bottleneck of label scarcity, recent works propose to incorporate few-shot learning frameworks for fast adaptations to graph classes with limited labeled graphs. Specifical…

2022

Few-Shot Learning on Graphs

IJCAI 2022poster

Graph representation learning has attracted tremendous attention due to its remarkable performance in many real-world applications. However, prevailing supervised graph representation learning models for specific tasks often suffer from label sparsity issue as data labeling is always time and resour…

Cited by 54SourcePDFScholar
2022

KCD: Knowledge Walks and Textual Cues Enhanced Political Perspective Detection in News Media

NAACL 2022long

Political perspective detection has become an increasingly important task that can help combat echo chambers and political polarization. Previous approaches generally focus on leveraging textual content to identify stances, while they fail to reason with background knowledge or leverage the rich sem…

2022

TwiBot-22: Towards Graph-Based Twitter Bot Detection

NeurIPS 2022accept

Twitter bot detection has become an increasingly important task to combat misinformation, facilitate social media moderation, and preserve the integrity of the online discourse. State-of-the-art bot detection methods generally leverage the graph structure of the Twitter network, and they exhibit pro…

2021

Multi-Cause Effect Estimation with Disentangled Confounder Representation

IJCAI 2021poster

One fundamental problem in causality learning is to estimate the causal effects of one or multiple treatments (e.g., medicines in the prescription) on an important outcome (e.g., cure of a disease). One major challenge of causal effect estimation is the existence of unobserved confounders -- the uno…

Cited by 15SourcePDFScholar
2020

IGNITE: A Minimax Game Toward Learning Individual Treatment Effects from Networked Observational Data

IJCAI 2020poster

Networked observational data presents new opportunities for learning individual causal effects, which plays an indispensable role in decision making. Such data poses the challenge of confounding bias. Previous work presents two desiderata to handle confounding bias. On the treatment group level, we…

Cited by 0SourcePDFScholar