← Search

Juntao Li

70 accepted papers

2026

$\textbf{Re}^{2}$: Unlocking LLM Reasoning via Reinforcement Learning with Re-solving

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has shown promise in enhancing the reasoning performance of large language models (LLMs) by increasing test-time compute. However, even after extensive RLVR training, such models still tend to generate unnecessary and low-quality steps in their…

Cited by 0SourceScholar
2026

CGMIS: Concept-Graph Based Multi-Hop Instructions Synthesis for Enhancing Long-Context Reasoning

AAAI 2026technical

High-quality multi-hop instruction data is critical for enhancing the reasoning capabilities of large language models (LLMs) in complex long-context scenarios, e.g., long-form reasoning. Nevertheless, there is currently a notable scarcity of such datasets within the community, and existing data synt

Cited by 0SourcePDFScholar
2026

Elastic Attention: Test-time Adaptive Sparsity Ratios for Efficient Transformers

ICML 2026poster

The quadratic complexity of standard attention mechanisms poses a significant scalability bottleneck for large language models (LLMs) in long-context scenarios. While hybrid attention strategies that combine sparse and full attention within a single model offer a viable solution, they typically empl…

Cited by 0SourceScholar
2026

FAPO: Flawed-Aware Policy Optimization for Efficient and Reliable Reasoning

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has emerged as a promising paradigm for enhancing the reasoning capabilities of large language models (LLMs). In this context, models explore reasoning trajectories and exploit rollouts with correct answers as positive signals for policy optimiz…

Cited by 0SourceScholar
2026

Revisiting Long-context Modeling from Context Denoising Perspective

ICLR 2026poster

Long-context models (LCMs) have demonstrated great potential in processing long sequences, facilitating many real-world applications. The success of LCMs can be attributed to their ability to locate implicit critical information within the context for further prediction. However, recent research rev…

Cited by 0SourcecodeScholar
2025

A Survey of Generative Information Extraction

COLING 2025main

Generative information extraction (Generative IE) aims to generate structured text sequences from unstructured text using a generative framework. Scaling in model size yields variations in adaptation and generalization, and also drives fundamental shifts in the techniques and approaches used within…

Cited by 1SourcePDFScholar
2025

ALW: Adaptive Layer-Wise contrastive decoding enhancing reasoning ability in Large Language Models

ACL 2025finding

Large language models (LLMs) have achieved remarkable performance across various reasoning tasks. However, many LLMs still encounter challenges in reasoning, especially for LLMs with fewer parameters or insufficient pre-training data. Through our experiments, we identify that noise accumulation acro…

2025

Accurate KV Cache Quantization with Outlier Tokens Tracing

ACL 2025long

The impressive capabilities of Large Language Models (LLMs) come at the cost of substantial computational resources during deployment. While KV Cache can significantly reduce recomputation during inference, it also introduces additional memory overhead. KV Cache quantization presents a promising sol…

2025

Alignment-Augmented Speculative Decoding with Alignment Sampling and Conditional Verification

EMNLP 2025

Recent works have revealed the great potential of speculative decoding in accelerating the autoregressive generation process of large language models. The success of these methods relies on the alignment between draft candidates and the sampled outputs of the target model. Existing methods mainly ac

2025

An Empirical Study of Iterative Refinements for Non-autoregressive Translation

ACL 2025long

Iterative non-autoregressive (NAR) models share a spirit of mixed autoregressive (AR) and fully NAR models, seeking a balance between generation quality and inference efficiency. These models have recently demonstrated impressive performance in varied generation tasks, surpassing the autoregressive…

2025

Beware of Calibration Data for Pruning Large Language Models

ICLR 2025poster

As large language models (LLMs) are widely applied across various fields, model compression has become increasingly crucial for reducing costs and improving inference efficiency. Post-training pruning is a promising method that does not require resource-intensive iterative training and only needs a…

Cited by 1SourcePDFScholar
2025

Decoder-Only LLMs can be Masked Auto-Encoders

ACL 2025short

Modern NLP workflows (e.g., RAG systems) require different models for generation and embedding tasks, where bidirectional pre-trained encoders and decoder-only Large Language Models (LLMs) dominate respective tasks. Structural differences between models result in extra development costs and limit kn…

2025

From Awareness to Adaptability: Enhancing Tool Utilization for Scientific Reasoning

ACL 2025finding

As large language models (LLMs) are increasingly applied to complex scientific problem-solving, their effectiveness is often limited by unconscious or failed tool usage. To address this issue, we introduce the Tool-Awareness Training (TAT) method, designed to enhance scientific reasoning. This appro…

2025

Generative Reward Modeling via Synthetic Criteria Preference Learning

ACL 2025long

Generative Reward Models (GenRMs) leverage synthesized Chains of Thought (CoT) to reduce the need for massive labeled data, but this approach introduces risks of overoptimization due to the inability to guarantee the correctness of the CoTs. Identifying and optimizing unexpected behaviors within the…

2025

Improving Rationality in the Reasoning Process of Language Models through Self-playing Game

ICML 2025poster

Large language models (LLMs) have demonstrated considerable reasoning abilities in various tasks such as mathematics and coding. However, recent studies indicate that even the best models lack true comprehension of their reasoning processes. In this paper, we explore how self-play can enhance the ra…

Cited by 0SourcePDFScholar
2025

L-CiteEval: A Suite for Evaluating Fidelity of Long-context Models

ACL 2025long

Long-context models(LCMs) have witnessed remarkable advancements in recent years, facilitating real-world tasks like long-document QA. The success of LCMs is founded on the hypothesis that the model demonstrates strong fidelity, enabling it to respond based on the provided long context rather than r…

Cited by 0SourcePDFScholar
2025

LOGO --- Long cOntext aliGnment via efficient preference Optimization

ICML 2025poster

Long-context models (LCMs) have shown great potential in processing long input sequences (even more than 100M tokens) conveniently and effectively. With significant progress, recent research has pointed out that LCMs can accurately locate token-level salient information within the context. Yet, the…

2025

MMA: Cross-Domain Knowledge Integration via Mixture of Multi-Domain Agents

EMNLP 2025

Rather than merely to retain previously acquired generalization, achieving synergistic improvements between generalization and domain specialization in foundation models remains a significant challenge in both pre-training and post-training. As an alternative, we propose a test-time cross-domain kno

2025

Revealing and Mitigating Over-Attention in Knowledge Editing

ICLR 2025poster

Large Language Models~(LLMs) have demonstrated superior performance across a wide range of tasks, but they still exhibit undesirable errors due to incorrect knowledge learned from the training data. To avoid this, knowledge editing methods emerged to precisely edit the specific model knowledge via e…

2025

Revealing and Mitigating the Local Pattern Shortcuts of Mamba

ACL 2025finding

Large language models (LLMs) have advanced significantly due to the attention mechanism, but their quadratic complexity and linear memory demands limit their performance on long-context tasks. Recently, researchers introduced Mamba, an advanced model built upon State Space Models (SSMs) that offers…

2025

SCAN: Self-Denoising Monte Carlo Annotation for Robust Process Reward Learning

NeurIPS 2025poster

Process reward models (PRMs) offer fine-grained, step-level evaluations that facilitate deeper reasoning processes in large language models (LLMs), proving effective in complex tasks like mathematical reasoning. However, developing PRMs is challenging due to the high cost and limited scalability of…

Cited by 0SourceScholar
2025

Scaling Code-Assisted Chain-of-Thoughts and Instructions for Model Reasoning

NeurIPS 2025poster

Reasoning capability is pivotal for Large Language Models (LLMs) to solve complex tasks, yet achieving reliable and scalable reasoning remains challenging. While Chain-of-Thought (CoT) prompting has become a mainstream approach, existing methods often suffer from uncontrolled generation, insufficien…

Cited by 0SourceScholar
2025

Thoughts Are All Over the Place: On the Underthinking of Long Reasoning Models

NeurIPS 2025spotlight

Long reasoning models (LRMs) such as OpenAI's o1 and DeepSeek's R1 have demonstrated remarkable abilities in complex reasoning tasks by scaling test-time compute and exhibiting human-like deep thinking. However, we identify a phenomenon we term underthinking, where LRMs frequently switch between dif…

Cited by 0SourcecodeScholar
2025

Tool learning via Inference-time Scaling and Cycle Verifier

ACL 2025finding

In inference-time scaling, Chain-of-Thought (CoT) plays a crucial role in enabling large language models (LLMs) to exhibit reasoning capabilities. However, in many scenarios, high-quality CoT data is scarce or even unavailable. In such cases, STaR-like methods can help LLMs synthesize CoT based on u…

2025

Unleashing LLM Reasoning Capability via Scalable Question Synthesis from Scratch

ACL 2025long

Improving the mathematical reasoning capabilities of Large Language Models (LLMs) is critical for advancing artificial intelligence. However, access to extensive, diverse, and high-quality reasoning datasets remains a significant challenge, particularly for the open-source community. In this paper,…

Cited by 0SourcePDFScholar
2025

Unlocking Recursive Thinking of LLMs: Alignment via Refinement

ACL 2025finding

The OpenAI o1-series models have demonstrated that leveraging long-form Chain of Thought (CoT) can substantially enhance performance. However, the recursive thinking capabilities of Large Language Models (LLMs) remain limited, particularly in the absence of expert-curated data for distillation. In t…

2025

Unveiling the Potential of BERT-family: A New Recipe for Building Scalable, General and Competitive Large Language Models

ACL 2025long

BERT-family have been increasingly explored for adaptation to scenarios beyond language understanding tasks, with more recent efforts focused on enabling them to become good instruction followers. These explorations have endowed BERT-family with new roles and human expectations, showcasing their pot…

2025

XIFBench: Evaluating Large Language Models on Multilingual Instruction Following

NeurIPS 2025poster

Large Language Models (LLMs) have demonstrated remarkable instruction-following capabilities across various applications. However, their performance in multilingual settings lacks systematic investigation, with existing evaluations lacking fine-grained constraint analysis across diverse linguistic c…

Cited by 0SourcecodeScholar
2025

𝒜3: Automatic Alignment Framework for Attributed Text Generation

ACL 2025long

Attributed text generation aims to enhance the reliability of content generated from large language models by providing citations for each claim, which thereby enables users to easily verify the correctness of the responses.However, the scarcity of high-quality training samples presents a significan…

2024

$\texttt{ConflictBank}$: A Benchmark for Evaluating the Influence of Knowledge Conflicts in LLMs

NeurIPS 2024poster

Large language models (LLMs) have achieved impressive advancements across numerous disciplines, yet the critical issue of knowledge conflicts, a major source of hallucinations, has rarely been studied. While a few research explored the conflicts between the inherent knowledge of LLMs and the retriev…

2024

Achieving Stronger Generation via Simple Contrastive Tuning

EMNLP 2024finding

Instruction tuning is widely used to unlock the abilities of Large Language Models (LLMs) in following human instructions, resulting in substantial performance improvements across various downstream tasks.Furthermore, contrastive decoding methods are employed to enhance instruction-tuned models. To…

2024

Adaptive Feature-based Low-Rank Compression of Large Language Models via Bayesian Optimization

EMNLP 2024finding

In recent years, large language models (LLMs) have driven advances in natural language processing. Still, their growing scale has increased the computational burden, necessitating a balance between efficiency and performance. Low-rank compression, a promising technique, reduces non-essential paramet…

2024

Are Bert Family Good Instruction Followers? A Study on Their Potential And Limitations

ICLR 2024poster

Language modeling at scale has proven very effective and brought unprecedented success to natural language models. Many typical representatives, especially decoder-only models, e.g., BLOOM and LLaMA, and encoder-decoder models, e.g., Flan-T5 and AlexaTM, have exhibited incredible instruction-followi…

Cited by 2SourcePDFScholar
2024

CMD: a framework for Context-aware Model self-Detoxification

EMNLP 2024main

Text detoxification aims to minimize the risk of language models producing toxic content. Existing detoxification methods of directly constraining the model output or further training the model on the non-toxic corpus fail to achieve a decent balance between detoxification effectiveness and generati…

2024

Demonstration Augmentation for Zero-shot In-context Learning

ACL 2024findings

Large Language Models (LLMs) have demonstrated an impressive capability known as In-context Learning (ICL), which enables them to acquire knowledge from textual demonstrations without the need for parameter updates.However, many studies have highlighted that the model’s performance is sensitive to t…

2024

Efficient Domain Adaptation for Non-Autoregressive Machine Translation

ACL 2024findings

Domain adaptation remains a challenge in the realm of Neural Machine Translation (NMT), even in the era of large language models (LLMs). Existing non-parametric approaches like nearest neighbor machine translation have made small Autoregressive Translation (AT) models achieve efficient domain genera…

2024

Exploring Reversal Mathematical Reasoning Ability for Large Language Models

ACL 2024findings

Large language models (LLMs) have presented remarkable capabilities in the wide range of natural language understanding and reasoning tasks. Despite their success, a few works indicate that LLMs suffer from the “reversal curse”, in which LLMs can’t employ the inverted structure “B is A” when they ar…

2024

Exploring and Mitigating Shortcut Learning for Generative Large Language Models

COLING 2024main

Recent generative large language models (LLMs) have exhibited incredible instruction-following capabilities while keeping strong task completion ability, even without task-specific fine-tuning. Some works attribute this to the bonus of the new scaling law, in which the continuous improvement of mode…

Cited by 7SourcePDFScholar
2024

LayoutNUWA: Revealing the Hidden Layout Expertise of Large Language Models

ICLR 2024poster

Graphic layout generation, a growing research field, plays a significant role in user engagement and information perception. Existing methods primarily treat layout generation as a numerical optimization task, focusing on quantitative aspects while overlooking the semantic information of layout, su…

2024

Living in the Moment: Can Large Language Models Grasp Co-Temporal Reasoning?

ACL 2024long

Temporal reasoning is fundamental for large language models (LLMs) to comprehend the world. Current temporal reasoning datasets are limited to questions about single or isolated events, falling short in mirroring the realistic temporal characteristics involving concurrent nature and intricate tempor…

2024

Rethinking Negative Instances for Generative Named Entity Recognition

ACL 2024findings

Large Language Models (LLMs) have demonstrated impressive capabilities for generalizing in unseen tasks. In the Named Entity Recognition (NER) task, recent advancements have seen the remarkable improvement of LLMs in a broad range of entity domains via instruction tuning, by adopting entity-centric…

2024

Retrieval and Reasoning on KGs: Integrate Knowledge Graphs into Large Language Models for Complex Question Answering

EMNLP 2024finding

Despite Large Language Models (LLMs) have performed impressively in various Natural Language Processing (NLP) tasks, their inherent hallucination phenomena severely challenge their credibility in complex reasoning. Combining explainable Knowledge Graphs (KGs) with LLMs is a promising path to address…

Cited by 7SourcePDFScholar
2024

StrokeNUWA—Tokenizing Strokes for Vector Graphic Synthesis

ICML 2024poster

To leverage LLMs for visual synthesis, traditional methods convert raster image information into discrete grid tokens through specialized visual modules, while disrupting the model’s ability to capture the true semantic representation of visual scenes. This paper posits that an alternative represent…

Cited by 12SourcePDFScholar
2024

Towards More Realistic Chinese Spell Checking with New Benchmark and Specialized Expert Model

COLING 2024main

Large Language Models (LLMs) hold considerable promise for artificial general intelligence, given their intrinsic abilities to accomplish a wide range of open-domain tasks either independently or in tandem with specialized expert models. However, despite these capabilities, the performance of LLMs h…

2023

AMOM: Adaptive Masking over Masking for Conditional Masked Language Model

AAAI 2023technical

Transformer-based autoregressive (AR) methods have achieved appealing performance for varied sequence-to-sequence generation tasks, e.g., neural machine translation, summarization, and code generation, but suffer from low inference efficiency. To speed up the inference stage, many non-autoregressive…

2023

AR-Diffusion: Auto-Regressive Diffusion Model for Text Generation

NeurIPS 2023poster

Diffusion models have gained significant attention in the realm of image generation due to their exceptional performance. Their success has been recently expanded to text generation via generating all tokens within a sequence concurrently. However, natural language exhibits a far more pronounced se…

2023

Beware of Model Collapse! Fast and Stable Test-time Adaptation for Robust Question Answering

EMNLP 2023long main

Although pre-trained language models (PLM) have achieved great success in question answering (QA), their robustness is still insufficient to support their practical applications, especially in the face of distribution shifts. Recently, test-time adaptation (TTA) has shown great potential for solving…

Cited by 0SourceScholar
2023

CORE: Cooperative Training of Retriever-Reranker for Effective Dialogue Response Selection

ACL 2023long

Establishing retrieval-based dialogue systems that can select appropriate responses from the pre-built index has gained increasing attention. Recent common practice is to construct a two-stage pipeline with a fast retriever (e.g., bi-encoder) for first-stage recall followed by a smart response reran…

Cited by 5SourcePDFScholar
2023

Can Diffusion Model Achieve Better Performance in Text Generation ? Bridging the Gap between Training and Inference !

ACL 2023findings

Diffusion models have been successfully adapted to text generation tasks by mapping the discrete text into the continuous space. However, there exist nonnegligible gaps between training and inference, owing to the absence of the forward process during inference. Thus, the model only predicts based o…

2023

Dynamic and Efficient Inference for Text Generation via BERT Family

ACL 2023long

Despite the excellent performance of Pre-trained Language Models on many text generation tasks, they suffer from inefficient inference on computation and memory due to their large-scale parameters and the universal autoregressive decoding paradigm. In this work, we propose a novel fine-tuning method…

2023

Early Exit with Disentangled Representation and Equiangular Tight Frame

ACL 2023findings

Dynamic early exit has demonstrated great potential in coping with the sharply increasing number of pre-trained language model parameters, which can achieve a good trade-off between performance and efficiency. The existing early exit paradigm relies on training parametrical internal classifiers at e…

2023

Efficient Continue Training of Temporal Language Model with Structural Information

EMNLP 2023long findings

Current language models are mainly trained on snap-shots of data gathered at a particular time, which decreases their capability to generalize over time and model language change. To model the \textit{time} variable, existing works have explored temporal language models (e.g., TempoBERT) by directl…

Cited by 0SourcecodeScholar
2023

G-SPEED: General SParse Efficient Editing MoDel

EMNLP 2023long findings

Large Language Models (LLMs) have demonstrated incredible capabilities in understanding, generating, and manipulating languages. Through human-model interactions, LLMs can automatically understand human-issued instructions and output the expected contents, which can significantly increase working ef…

Cited by 0SourcecodeScholar
2023

INFORM : Information eNtropy based multi-step reasoning FOR large language Models

EMNLP 2023long main

Large language models (LLMs) have demonstrated exceptional performance in reasoning tasks with dedicated Chain-of-Thought (CoT) prompts. Further enhancing CoT prompts with exquisite exemplars can significantly improve reasoning performance.However, the effectiveness of CoT prompts may fluctuate dram…

Cited by 0SourceScholar
2023

Isotropic Representation Can Improve Zero-Shot Cross-Lingual Transfer on Multilingual Language Models

EMNLP 2023long findings

With the development of multilingual pre-trained language models (mPLMs), zero-shot cross-lingual transfer shows great potential. To further improve the performance of cross-lingual transfer, many studies have explored representation misalignment caused by morphological differences but neglected the…

Cited by 0SourcecodeScholar
2023

KBioXLM: A Knowledge-anchored Biomedical Multilingual Pretrained Language Model

EMNLP 2023long findings

Most biomedical pretrained language models are monolingual and cannot handle the growing cross-lingual requirements. The scarcity of non-English domain corpora, not to mention parallel data, poses a significant hurdle in training multilingual biomedical models. Since knowledge forms the core of doma…

Cited by 0SourcecodeScholar
2023

Open-ended Long Text Generation via Masked Language Modeling

ACL 2023long

Pre-trained autoregressive (AR) language models such as BART and GPTs have dominated OPen-ended Long Text Generation (Open-LTG).However, the AR nature will decrease the inference efficiency along with the increase of generation length, which hinder their application in Open-LTG.To improve inference…

2023

RenewNAT: Renewing Potential Translation for Non-autoregressive Transformer

AAAI 2023technical

Non-autoregressive neural machine translation (NAT) models are proposed to accelerate the inference process while maintaining relatively high performance. However, existing NAT models are difficult to achieve the desired efficiency-quality trade-off. For one thing, fully NAT models with efficient in…

Cited by 6SourcePDFScholar
2023

Towards Better Hierarchical Text Classification with Data Generation

ACL 2023findings

Hierarchical text classification (HTC) focuses on classifying one text into multiple labels, which are organized as a hierarchical taxonomy. Due to its wide involution in realistic scenarios, HTC attracts long-term attention from both industry and academia. However, the high cost of hierarchical mul…

2022

Gated Mechanism Enhanced Multi-Task Learning for Dialog Routing

COLING 2022main

Currently, human-bot symbiosis dialog systems, e.g. pre- and after-sales in E-commerce, are ubiquitous, and the dialog routing component is essential to improve the overall efficiency, reduce human resource cost and increase user experience. To satisfy this requirement, existing methods are mostly h…

Cited by 0SourcePDFScholar
2022

Image-text Retrieval: A Survey on Recent Research and Development

IJCAI 2022poster

In the past few years, cross-modal image-text retrieval (ITR) has experienced increased interest in the research community due to its excellent research value and broad real-world application. It is designed for the scenarios where the queries are from one modality and the retrieval galleries from a…

Cited by 112SourcePDFScholar
2022

Improving Temporal Generalization of Pre-trained Language Models with Lexical Semantic Change

EMNLP 2022main

Recent research has revealed that neural language models at scale suffer from poor temporal generalization capability, i.e., language model pre-trained on static data from past years performs worse over time on emerging data. Existing methods mainly perform continual training to mitigate such a misa…

2022

JANUS: Joint Autoregressive and Non-autoregressive Training with Auxiliary Loss for Sequence Generation

EMNLP 2022main

Transformer-based autoregressive and non-autoregressive models have played an essential role in sequence generation tasks. The autoregressive model can obtain excellent performance, while the non-autoregressive model brings fast decoding speed for inference. In this paper, we propose JANUS, a Joint…

2022

Robust Question Answering against Distribution Shifts with Test-Time Adaption: An Empirical Study

EMNLP 2022finding

A deployed question answering (QA) model can easily fail when the test data has a distribution shift compared to the training data. Robustness tuning (RT) methods have been widely studied to enhance model robustness against distribution shifts before model deployment. However, can we improve a model…

Cited by 12SourcePDFScholar
2022

SelfMix: Robust Learning against Textual Label Noise with Self-Mixup Training

COLING 2022main

The conventional success of textual classification relies on annotated data, and the new paradigm of pre-trained language models (PLMs) still requires a few labeled data for downstream tasks. However, in real-world applications, label noise inevitably exists in training data, damaging the effectiven…

2021

Content Learning with Structure-Aware Writing: A Graph-Infused Dual Conditional Variational Autoencoder for Automatic Storytelling

AAAI 2021technical

Recent automatic storytelling methods mainly rely on keyword planning or plot skeleton generation to model long-range dependencies and create consistent narrative texts. However, these approaches generate story plans or plots sequentially, leaving the non-sequential conception and structural design…

Cited by 11SourcePDFScholar
2021

Learning to Organize a Bag of Words into Sentences with Neural Networks: An Empirical Study

NAACL 2021long

Sequential information, a.k.a., orders, is assumed to be essential for processing a sequence with recurrent neural network or convolutional neural network based encoders. However, is it possible to encode natural languages without orders? Given a bag of words from a disordered sentence, humans may s…

Cited by 9SourcePDFScholar
2021

R-Drop: Regularized Dropout for Neural Networks

NeurIPS 2021poster

Dropout is a powerful and widely used technique to regularize the training of deep neural networks. Though effective and performing well, the randomness introduced by dropout causes unnegligible inconsistency between training and inference. In this paper, we introduce a simple consistency training s…

2020

Unsupervised Domain Adaptation of a Pretrained Cross-Lingual Language Model

IJCAI 2020poster

Recent research indicates that pretraining cross-lingual language models on large-scale unlabeled texts yields significant performance improvements over various cross-lingual and low-resource tasks. Through training on one hundred languages and terabytes of texts, cross-lingual language models have…