← Search

Yifan Song

29 accepted papers

2026

Learning To Draft: Adaptive Speculative Decoding with Reinforcement Learning

ICLR 2026poster

Speculative decoding accelerates large language model (LLM) inference by using a small draft model to generate candidate tokens for a larger target model to verify. The efficacy of this technique hinges on the trade-off between the time spent on drafting candidates and verifying them. However, curre…

Cited by 0SourcecodeScholar
2026

Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers

ICML 2026poster

Reinforcement learning (RL) has emerged as a crucial approach for enhancing the capabilities of large language models. However, in Mixture-of-Experts (MoE) models, the routing mechanism often introduces instability, even leading to catastrophic RL training collapse. We analyze the training-inference…

Cited by 0SourceScholar
2025

Adding Additional Control to One-Step Diffusion with Joint Distribution Matching

ICCV 2025poster

While diffusion distillation has enabled one-step generation through methods like Variational Score Distillation, adapting distilled models to emerging *new controls* -- such as novel structural constraints or latest user preferences -- remains challenging. Conventional approaches typically requires…

Cited by 0SourcePDFScholar
2025

Decoupled Graph Energy-based Model for Node Out-of-Distribution Detection on Heterophilic Graphs

ICLR 2025poster

Despite extensive research efforts focused on Out-of-Distribution (OOD) detection on images, OOD detection on nodes in graph learning remains underexplored. The dependence among graph nodes hinders the trivial adaptation of existing approaches on images that assume inputs to be i.i.d. sampled, since…

2025

Exploring Fine-Grained Human Motion Video Captioning

COLING 2025main

Detailed descriptions of human motion are crucial for effective fitness training, which highlights the importance of research in fine-grained human motion video captioning. Existing video captioning models often fail to capture the nuanced semantics of videos, resulting in the generated descriptions…

2025

Harnessing Webpage UIs for Text-Rich Visual Understanding

ICLR 2025poster

Text-rich visual understanding—the ability to interpret both textual content and visual elements within a scene—is crucial for multimodal large language models (MLLMs) to effectively interact with structured environments. We propose leveraging webpage UIs as a naturally structured and diverse data s…

Cited by 3SourcePDFScholar
2025

Hierarchical Memory Organization for Wikipedia Generation

ACL 2025long

Generating Wikipedia articles autonomously is a challenging task requiring the integration of accurate, comprehensive, and well-structured information from diverse sources. This paper introduces the Memory Organization-based Generation (MOG) framework, a novel approach to address these challenges by…

2025

ISR: Self-Refining Referring Expressions for Entity Grounding

ACL 2025long

Entity grounding, a crucial task in constructing multimodal knowledge graphs, aims to align entities from knowledge graphs with their corresponding images. Unlike conventional visual grounding tasks that use referring expressions (REs) as inputs, entity grounding relies solely on entity names and ty…

2025

MPO: Boosting LLM Agents with Meta Plan Optimization

EMNLP 2025

Recent advancements in large language models (LLMs) have enabled LLM-based agents to successfully tackle interactive planning tasks. However, despite their successes, existing approaches often suffer from planning hallucinations and require retraining for each new agent. To address these challenges,

2025

MixEval-X: Any-to-any Evaluations from Real-world Data Mixture

ICLR 2025spotlight

Perceiving and generating diverse modalities are crucial for AI models to effectively learn from and engage with real-world signals, necessitating reliable evaluations for their development. We identify two major issues in current evaluations: (1) inconsistent standards, shaped by different communit…

Cited by 3SourcePDFScholar
2025

More Tokens, Lower Precision: Towards the Optimal Token-Precision Trade-off in KV Cache Compression

EMNLP 2025

As large language models (LLMs) process increasing context windows, the memory usage of KV cache has become a critical bottleneck during inference. The mainstream KV compression methods, including KV pruning and KV quantization, primarily focus on either token or precision dimensions separately. How

2025

The Good, The Bad, and The Greedy: Evaluation of LLMs Should Not Ignore Non-Determinism

NAACL 2025long

Current evaluations of large language models (LLMs) often overlook non-determinism, typically focusing on a single output per example. This limits our understanding of LLM performance variability in real-world applications. Our study addresses this issue by exploring key questions about the performa…

2025

VL-RewardBench: A Challenging Benchmark for Vision-Language Generative Reward Models

CVPR 2025highlight

Vision-language generative reward models (VL-GenRMs) play a crucial role in aligning and evaluating multimodal AI systems, yet their own evaluation remains under-explored. Current assessment methods primarily rely on AI-annotated preference labels from traditional VL tasks, which can introduce biase…

2024

A Learning-Based Multi-Node Fusion Positioning Method Using Wearable Inertial Sensors

ICASSP 2024accepted

This study presents a novel approach to enhance the accuracy and adaptability of pedestrian positioning by fusing data from multiple Inertial Measurement Units (IMUs) attached to the human body. Leveraging the temporal and spatial richness of IMU data, our proposed multi-node sensors fusion strategy…

Cited by 0SourceScholar
2024

AgentBank: Towards Generalized LLM Agents via Fine-Tuning on 50000+ Interaction Trajectories

EMNLP 2024finding

Fine-tuning on agent-environment interaction trajectory data holds significant promise for surfacing generalized agent capabilities in open-source large language models (LLMs). In this work, we introduce AgentBank, by far the largest trajectory tuning data collection featuring more than 50k diverse…

2024

CoUDA: Coherence Evaluation via Unified Data Augmentation

NAACL 2024long

Coherence evaluation aims to assess the organization and structure of a discourse, which remains challenging even in the era of large language models. Due to the scarcity of annotated data, data augmentation is commonly used for training coherence evaluation models. However, previous augmentations f…

2024

LongEmbed: Extending Embedding Models for Long Context Retrieval

EMNLP 2024main

Embedding models play a pivotal role in modern NLP applications such as document retrieval. However, existing embedding models are limited to encoding short documents of typically 512 tokens, restrained from application scenarios requiring long inputs. This paper explores context window extension of…

2024

PoSE: Efficient Context Window Extension of LLMs via Positional Skip-wise Training

ICLR 2024poster

Large Language Models (LLMs) are trained with a pre-defined context length, restricting their use in scenarios requiring long inputs. Previous efforts for adapting LLMs to a longer length usually requires fine-tuning with this target length (Full-length fine-tuning), suffering intensive training cos…

2024

Trial and Error: Exploration-Based Trajectory Optimization of LLM Agents

ACL 2024long

Large Language Models (LLMs) have become integral components in various autonomous agent systems.In this study, we present an exploration-based trajectory optimization approach, referred to as ETO. This learning method is designed to enhance the performance of open LLM agents. Contrary to previous s…

2024

Watch Every Step! LLM Agent Learning via Iterative Step-level Process Refinement

EMNLP 2024main

Large language model agents have exhibited exceptional performance across a range of complex interactive tasks. Recent approaches have utilized tuning with expert trajectories to enhance agent performance, yet they primarily concentrate on outcome rewards, which may lead to errors or suboptimal acti…

2024

mmBaT: A Multi-Task Framework for Mmwave-Based Human Body Reconstruction and Translation Prediction

ICASSP 2024accepted

Human body reconstruction with Millimeter Wave (mmWave) radar point clouds has gained significant interest due to its ability to work in adverse environments and its capacity to mitigate privacy concerns associated with traditional camera-based solutions. Despite pioneering efforts in this field, tw…

Cited by 0SourceScholar
2023

DocRED-FE: A Document-Level Fine-Grained Entity and Relation Extraction Dataset

ICASSP 2023accepted

Joint entity and relation extraction (JERE) is one of the most important tasks in information extraction. However, most existing works focus on sentence-level coarse-grained JERE, which have limitations in real-world scenarios. In this paper, we construct a large-scale document-level fine-grained JE…

Cited by 0SourceScholar
2023

InfoCL: Alleviating Catastrophic Forgetting in Continual Text Classification from An Information Theoretic Perspective

EMNLP 2023long findings

Continual learning (CL) aims to constantly learn new knowledge over time while avoiding catastrophic forgetting on old tasks. We focus on continual text classification under the class-incremental setting. Recent CL studies have identified the severe performance decrease on analogous classes as a key…

Cited by 0SourcecodeScholar
2023

Rationale-Enhanced Language Models are Better Continual Relation Learners

EMNLP 2023short main

Continual relation extraction (CRE) aims to solve the problem of catastrophic forgetting when learning a sequence of newly emerging relations. Recent CRE studies have found that catastrophic forgetting arises from the model's lack of robustness against future analogous relations. To address the issu…

Cited by 0SourcecodeScholar
2022

Calibrating Factual Knowledge in Pretrained Language Models

EMNLP 2022finding

Previous literature has proved that Pretrained Language Models (PLMs) can store factual knowledge. However, we find that facts stored in the PLMs are not always correct. It motivates us to explore a fundamental question: How do we calibrate factual knowledge in PLMs without re-training from scratch?…

2022

ConFiguRe: Exploring Discourse-level Chinese Figures of Speech

COLING 2022main

Figures of speech, such as metaphor and irony, are ubiquitous in literature works and colloquial conversations. This poses great challenge for natural language understanding since figures of speech usually deviate from their ostensible meanings to express deeper semantic implications. Previous resea…

2022

Learning Robust Representations for Continual Relation Extraction via Adversarial Class Augmentation

EMNLP 2022main

Continual relation extraction (CRE) aims to continually learn new relations from a class-incremental data stream. CRE model usually suffers from catastrophic forgetting problem, i.e., the performance of old relations seriously degrades when the model learns new relations. Most previous work attribut…

2022

Robust Fine-tuning via Perturbation and Interpolation from In-batch Instances

IJCAI 2022poster

Fine-tuning pretrained language models (PLMs) on downstream tasks has become common practice in natural language processing. However, most of the PLMs are vulnerable, e.g., they are brittle under adversarial attacks or imbalanced data, which hinders the application of the PLMs on some downstream tas…