← Search

Yihong Chen

18 accepted papers

2026

Jet Expansions: Restructuring LLM Computation for Model Inspection

ICLR 2026poster

Large language models are becoming general knowledge engines for diverse applications. However, their computations are deeply entangled after training, resisting modularization which complicates interpretability, auditing, and long-term maintenance. We introduce Jet Expansions, a framework for expan…

Cited by 0SourceScholar
2026

ReAugment: Targeted Few-Shot Time Series Augmentation via Model Zoo-Guided Reinforcement Learning

ICML 2026poster

Few-shot time series forecasting is fundamentally challenged by the scarcity of high-quality training data and the risk of severe overfitting. To address this issue, we propose ReAugment, a reinforcement learning (RL) framework that explicitly learns where and how to augment time series data. ReAugm…

Cited by 0SourceScholar
2026

Rethinking Data Curation in LLM Training: Online Reweighting Offers Better Generalization than Offline Methods

ICLR 2026poster

Data curation is a critical yet underexplored component in large language model (LLM) training. Existing approaches (such as data selection and data mixing) operate in an offline paradigm, decoupled from the training process. This separation introduces extra engineering overhead and makes curated su…

Cited by 0SourcecodeScholar
2025

LLM Unlearning via Neural Activation Redirection

NeurIPS 2025poster

The ability to selectively remove knowledge from LLMs is highly desirable. However, existing methods often struggle with balancing unlearning efficacy and retain model utility, and lack controllability at inference time to emulate base model behavior as if it had never seen the unlearned data. In th…

Cited by 0SourceScholar
2025

Multilingual Language Model Pretraining using Machine-translated Data

EMNLP 2025

English, as a very high-resource language, enables the pretraining of high-quality large language models (LLMs). However, the same can not be said for most other languages, likely due to a gap in the quality and diversity of available multilingual pretraining corpora. In this work, we find that docu

Cited by 0SourcePDFScholar
2024

AsEP: Benchmarking Deep Learning Methods for Antibody-specific Epitope Prediction

NeurIPS 2024poster

Epitope identification is vital for antibody design yet challenging due to the inherent variability in antibodies. While many deep learning methods have been developed for general protein binding site prediction tasks, whether they work for epitope prediction remains an understudied research questio…

2024

Breaking Physical and Linguistic Borders: Multilingual Federated Prompt Tuning for Low-Resource Languages

ICLR 2024poster

Pretrained large language models (LLMs) have emerged as a cornerstone in modern natural language processing, with their utility expanding to various applications and languages. However, the fine-tuning of multilingual LLMs, particularly for low-resource languages, is fraught with challenges steming…

Cited by 17SourcePDFScholar
2024

TPTU-v2: Boosting Task Planning and Tool Usage of Large Language Model-based Agents in Real-world Industry Systems

EMNLP 2024industry

Large Language Models (LLMs) have demonstrated proficiency in addressing tasks that necessitate a combination of task planning and the usage of external tools, such as weather and calculator APIs. However, real-world industrial systems present prevalent challenges in task planning and tool usage: nu…

2023

Improving Language Plasticity via Pretraining with Active Forgetting

NeurIPS 2023poster

Pretrained language models (PLMs) are today the primary model for natural language processing. Despite their impressive downstream performance, it can be difficult to apply PLMs to new languages, a barrier to making their capabilities universally accessible. While prior work has shown it possible to…

2023

Mini-Model Adaptation: Efficiently Extending Pretrained Models to New Languages via Aligned Shallow Training

ACL 2023findings

Prior work shows that it is possible to expand pretrained Masked Language Models (MLMs) to new languages by learning a new set of embeddings, while keeping the transformer body frozen. Despite learning a small subset of parameters, this approach is not compute-efficient, as training the new embeddin…

2022

Check and Link: Pairwise Lesion Correspondence Guides Mammogram Mass Detection

ECCV 2022poster

"Detecting mass in mammogram is significant due to the high occurrence and mortality of breast cancer. In mammogram mass detection, modeling pairwise lesion correspondence explicitly is particularly important. However, most of the existing methods build relatively coarse correspondence and have not…

Cited by 6SourcePDFScholar
2022

PointScatter: Point Set Representation for Tubular Structure Extraction

ECCV 2022poster

"This paper explores the point set representation for tubular structure extraction tasks. Compared with the traditional mask representation, the point set representation enjoys its flexibility and representation ability, which would not be restricted by the fixed grid as the mask. Inspired by this,…

2022

ReFactor GNNs: Revisiting Factorisation-based Models from a Message-Passing Perspective

NeurIPS 2022accept

Factorisation-based Models (FMs), such as DistMult, have enjoyed enduring success for Knowledge Graph Completion (KGC) tasks, often outperforming Graph Neural Networks (GNNs). However, unlike GNNs, FMs struggle to incorporate node features and generalise to unseen nodes in inductive settings. Our wo…

2020

RepPoints v2: Verification Meets Regression for Object Detection

NeurIPS 2020poster

Verification and regression are two general methodologies for prediction in neural networks. Each has its own strengths: verification can be easier to infer accurately, and regression is more efficient and applicable to continuous target variables. Hence, it is often beneficial to carefully combine…