← Search

Tianjian Li

9 accepted papers

2026

Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generations

ICML 2026poster

High-quality kernel is critical for scalable AI systems, and enabling LLMs to generate such code would advance AI development. However, training LLMs for this task requires sufficient data, a robust environment, and the process is often vulnerable to _reward hacking_ and _lazy optimization_. In thes…

Cited by 0SourceScholar
2026

Mini-o3: Scaling Up Reasoning Patterns and Interaction Turns for Visual Search

ICLR 2026poster

Recent advances in large multimodal models have leveraged image-based tools with reinforcement learning to tackle visual problems. However, existing open-source approaches often exhibit monotonous reasoning patterns and allow only a limited number of interaction turns, making them inadequate for dif…

Cited by 0SourcecodeScholar
2025

Benchmarking Language Model Creativity: A Case Study on Code Generation

NAACL 2025long

As LLMs become increasingly prevalent, it is interesting to consider how “creative” these models can be. From cognitive science, creativity consists of at least two key characteristics: convergent thinking (purposefulness to achieve a given goal) and divergent thinking (adaptability to explore new e…

2025

SIMPLEMIX: Frustratingly Simple Mixing of Off- and On-policy Data in Language Model Preference Learning

ICML 2025poster

Aligning language models with human preferences relies on pairwise preference datasets. While some studies suggest that on-policy data consistently outperforms off-policy data for preference learning, others indicate that the advantages of on-policy data are task-dependent, highlighting the need for…

Cited by 0SourcePDFScholar
2025

Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasets

NAACL 2025long

Data abundance across different domains exhibits a long-tailed distribution: few domains have abundant data, while most face data scarcity. Our work focuses on a multilingual setting, where available data is heavily skewed toward high-resource languages, creating significant imbalances in training d…

Cited by 0SourcePDFScholar
2025

Verifiable by Design: Aligning Language Models to Quote from Pre-Training Data

NAACL 2025long

To trust the fluent generations of large language models (LLMs), humans must be able to _verify_ their correctness against trusted, external sources. Recent efforts, such as providing citations via retrieved documents or post-hoc provenance, enhance verifiability but provide no guarantees on their c…

2025

ZeCO: Zero-Communication Overhead Sequence Parallelism for Linear Attention

NeurIPS 2025poster

Linear attention mechanisms deliver significant advantages for Large Language Models (LLMs) by providing linear computational complexity, enabling efficient processing of ultra-long sequences (e.g., 1M context). However, existing Sequence Parallelism (SP) methods, essential for distributing these wo…

Cited by 0SourceScholar
2024

Error Norm Truncation: Robust Training in the Presence of Data Noise for Text Generation Models

ICLR 2024spotlight

Text generation models are notoriously vulnerable to errors in the training data. With the wide-spread availability of massive amounts of web-crawled data becoming more commonplace, how can we enhance the robustness of models trained on a massive amount of noisy web-crawled text? In our work, we pro…

Cited by 4SourcePDFScholar
2023

Why Does Zero-Shot Cross-Lingual Generation Fail? An Explanation and a Solution

ACL 2023findings

Zero-shot cross-lingual transfer is when a multilingual model is trained to perform a task in one language and then is applied to another language. Although the zero-shot cross-lingual transfer approach has achieved success in various classification tasks, its performance on natural language generat…