← Search

Jiaqi W. Ma

8 accepted papers

2025

A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learning

NeurIPS 2025oral

Online reinforcement learning (RL) excels in complex, safety-critical domains but suffers from sample inefficiency, training instability, and limited interpretability. Data attribution provides a principled way to trace model behavior back to training samples, yet existing methods assume fixed datas…

Cited by 0SourcecodeScholar
2025

A Versatile Influence Function for Data Attribution with Non-Decomposable Loss

ICML 2025poster

Influence function, a technique rooted in robust statistics, has been adapted in modern machine learning for a novel application: data attribution---quantifying how individual training data points affect a model's predictions. However, the common derivation of influence functions in the data attribu…

Cited by 0SourcePDFScholar
2025

DATE-LM: Benchmarking Data Attribution Evaluation for Large Language Models

NeurIPS 2025poster

Data attribution methods quantify the influence of training data on model outputs and are becoming increasingly relevant for a wide range of LLM research and applications, including dataset curation, model interpretability, data valuation. However, there remain critical gaps in systematic LLM-centri…

Cited by 0SourceScholar
2025

GraSS: Scalable Data Attribution with Gradient Sparsification and Sparse Projection

NeurIPS 2025poster

Gradient-based data attribution methods, such as influence functions, are critical for understanding the impact of individual training samples without requiring repeated model retraining. However, their scalability is often limited by the high computational and memory costs associated with per-sampl…

Cited by 0SourcecodeScholar
2025

Improving Influence-based Instruction Tuning Data Selection for Balanced Learning of Diverse Capabilities

EMNLP 2025

Selecting appropriate training data is crucial for instruction fine-tuning of large language models (LLMs), which aims to (1) elicit strong capabilities, and (2) achieve balanced performance across different tasks. Influence-based methods show promise in achieving (1), by estimating the contribution

Cited by 0SourcePDFScholar
2025

Taming Hyperparameter Sensitivity in Data Attribution: Practical Selection Without Costly Retraining

NeurIPS 2025poster

Data attribution methods, which quantify the influence of individual training data points on a machine learning model, have gained increasing popularity in data-centric applications in modern AI. Despite a recent surge of new methods developed in this space, the impact of hyperparameter tuning in th…

Cited by 0SourcecodeScholar