← Search

SHIYANG LI

23 accepted papers

2026

Dual-Process Distribution Calibration: Bridging Slow-Fast Thinking for Few-Shot Learning

IJCAI 2026

Artificial intelligence models typically perform well on large-scale datasets, yet their effectiveness tends to degrade in real-world scenarios with scarce data, such as medical diagnostics. In contrast, humans can learn and reason effectively from few examples. Even when novel objects differ signif

Cited by 0Scholar
2026

StitchCUDA: An Automated Multi-Agents End-to-End GPU Programing Framework with Rubric-based Agentic Reinforcement Learning

ICML 2026poster

Modern machine learning (ML) workloads increasingly rely on GPUs, yet achieving high end-to-end performance remains challenging due to dependencies on both GPU kernel efficiency and host-side settings. Although LLM-based methods show promise on automated GPU kernel generation, prior works mainly foc…

Cited by 0SourceScholar
2025

Can Language Models Follow Multiple Turns of Entangled Instructions?

EMNLP 2025

Despite of significant achievements in improving instruction-following capabilities of large language models (LLMs), the ability to process multiple potentially entangled or conflict instructions remains a considerable challenge. Real-world scenarios often require the consistency across multiple ins

2025

DrAgent: Empowering Large Language Models as Medical Agents for Multi-hop Medical Reasoning

EMNLP 2025

Although large language models (LLMs) have demonstrated outperforming human experts in medical examinations, it remains challenging to adopt LLMs in real-world clinical decision-making that typically involves multi-hop medical reasoning. Common practices include prompting commercial LLMs and fine-tu

Cited by 0SourcePDFScholar
2025

IHEval: Evaluating Language Models on Following the Instruction Hierarchy

NAACL 2025long

The instruction hierarchy, which establishes a priority order from system messages to user messages, conversation history, and tool outputs, is essential for ensuring consistent and safe behavior in language models (LMs). Despite its importance, this topic receives limited attention, and there is a…

2025

Train a Unified Multimodal Data Quality Classifier with Synthetic Data

EMNLP 2025

The Multimodal Large Language Models (MLLMs) are continually pre-trained on a mixture of image-text caption data and interleaved document data, while the high-quality data filtering towards image-text interleaved document data is under-explored. We propose to train an efficient MLLM as a Unified Mul

Cited by 0SourcePDFScholar
2024

AlpaGasus: Training a Better Alpaca with Fewer Data

ICLR 2024poster

Large language models~(LLMs) strengthen instruction-following capability through instruction-finetuning (IFT) on supervised instruction/response data. However, widely used IFT datasets (e.g., Alpaca's 52k data) surprisingly contain many low-quality instances with incorrect or irrelevant responses, w…

2024

Backdooring Instruction-Tuned Large Language Models with Virtual Prompt Injection

NAACL 2024long

Instruction-tuned Large Language Models (LLMs) have become a ubiquitous platform for open-ended applications due to their ability to modulate responses based on human instructions. The widespread use of LLMs holds significant potential for shaping public perception, yet also risks being maliciously…

2024

Data Diversity Matters for Robust Instruction Tuning

EMNLP 2024finding

Recent works have shown that by curating high quality and diverse instruction tuning datasets, we can significantly improve instruction-following capabilities. However, creating such datasets is difficult and most works rely on manual curation or proprietary language models. Automatic data curation…

2024

Enhancing Small Medical Learners with Privacy-preserving Contextual Prompting

ICLR 2024poster

Large language models (LLMs) demonstrate remarkable medical expertise, but data privacy concerns impede their direct use in healthcare environments. Although offering improved data privacy protection, domain-specific small language models (SLMs) often underperform LLMs, emphasizing the need for meth…

2024

Instruction-following Evaluation through Verbalizer Manipulation

NAACL 2024findings

While instruction-tuned models have shown remarkable success in various natural language processing tasks, accurately evaluating their ability to follow instructions remains challenging. Existing benchmarks primarily focus on common instructions that align well with what the model learned during tra…

2024

MEMORYLLM: Towards Self-Updatable Large Language Models

ICML 2024poster

Existing Large Language Models (LLMs) usually remain static after deployment, which might make it hard to inject new knowledge into the model. We aim to build models containing a considerable portion of self-updatable parameters, enabling the model to integrate new knowledge effectively and efficien…

2023

Graph Reasoning for Question Answering with Triplet Retrieval

ACL 2023findings

Answering complex questions often requires reasoning over knowledge graphs (KGs). State-of-the-art methods often utilize entities in questions to retrieve local subgraphs, which are then fed into KG encoder, e.g. graph neural networks (GNNs), to model their local structures and integrated into langu…

2023

Improving Medical Predictions by Irregular Multimodal Electronic Health Records Modeling

ICML 2023poster

Health conditions among patients in intensive care units (ICUs) are monitored via electronic health records (EHRs), composed of numerical time series and lengthy clinical note sequences, both taken at $\textit{irregular}$ time intervals. Dealing with such irregularity in every modality, and integrat…

2023

Limitations of Language Models in Arithmetic and Symbolic Induction

ACL 2023long

Recent work has shown that large pretrained Language Models (LMs) can not only perform remarkably well on a range of Natural Language Processing (NLP) tasks but also start improving on reasoning tasks such as arithmetic induction, symbolic manipulation, and commonsense reasoning with increasing size…

Cited by 72SourcePDFScholar
2023

Time Series as Images: Vision Transformer for Irregularly Sampled Time Series

NeurIPS 2023poster

Irregularly sampled time series are increasingly prevalent, particularly in medical domains. While various specialized methods have been developed to handle these irregularities, effectively modeling their complex dynamics and pronounced sparsity remains a challenge. This paper introduces a novel p…

2022

Controllable Dialogue Simulation with In-context Learning

EMNLP 2022finding

Building dialogue systems requires a large corpus of annotated dialogues. Such datasets are usually created via crowdsourcing, which is expensive and time-consuming. In this paper, we propose Dialogic, a novel dialogue simulation method based on large language model in-context learning to automate d…

2022

ConvFinQA: Exploring the Chain of Numerical Reasoning in Conversational Finance Question Answering

EMNLP 2022main

With the recent advance in large pre-trained language models, researchers have achieved record performances in NLP tasks that mostly focus on language pattern matching. The community is experiencing the shift of the challenge from how to model language to the imitation of complex reasoning abilities…

2021

CoCo: Controllable Counterfactuals for Evaluating Dialogue State Trackers

ICLR 2021poster

Dialogue state trackers have made significant progress on benchmark datasets, but their generalization capability to novel and realistic scenarios beyond the held- out conversations is less understood. We propose controllable counterfactuals (COCO) to bridge this gap and evaluate dialogue state trac…

2021

Task-adaptive Pre-training and Self-training are Complementary for Natural Language Understanding

EMNLP 2021finding

Task-adaptive pre-training (TAPT) and Self-training (ST) have emerged as the major semi-supervised approaches to improve natural language understanding (NLU) tasks with massive amount of unlabeled data. However, it’s unclear whether they learn similar representations or they can be effectively combi…

Cited by 14SourcePDFScholar
2020

TabFact: A Large-scale Dataset for Table-based Fact Verification

ICLR 2020poster

The problem of verifying whether a textual hypothesis holds based on the given evidence, also known as fact verification, plays an important role in the study of natural language understanding and semantic representation. However, existing studies are mainly restricted to dealing with unstructured e…

Cited by 517SourcecodeScholar
2019

Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecasting

NeurIPS 2019poster

Time series forecasting is an important problem across many domains, including predictions of solar plant energy output, electricity consumption, and traffic jam situation. In this paper, we propose to tackle such forecasting problem with Transformer. Although impressed by its performance in our pre…

Cited by 2073SourcePDFScholar
2018

Towards Understanding Acceleration Tradeoff between Momentum and Asynchrony in Nonconvex Stochastic Optimization

NeurIPS 2018poster

Asynchronous momentum stochastic gradient descent algorithms (Async-MSGD) have been widely used in distributed machine learning, e.g., training large collaborative filtering systems and deep neural networks. Due to current technical limit, however, establishing convergence properties of Async-MSGD f…

Cited by 11SourcePDFScholar