← Search

Ruochen Jin

3 accepted papers

2026

RMNP: Row-Momentum Normalized Preconditioning for Scalable Matrix-Based Optimization

ICML 2026poster

Preconditioned adaptive methods have gained significant attention for training deep neural networks, as they capture rich curvature information. The central challenge in this field lies in balancing preconditioning effectiveness with computational efficiency of implementing the preconditioner. Among…

Cited by 0SourceScholar
2025

Fine-Tuning Attention Modules Only: Enhancing Weight Disentanglement in Task Arithmetic

ICLR 2025poster

In recent years, *task arithmetic* has garnered increasing attention. This approach edits pre-trained models directly in weight space by combining the fine-tuned weights of various tasks into a *unified model*. Its efficiency and cost-effectiveness stem from its training-free combination, contrastin…

Cited by 1SourcePDFScholar
2025

Restoring Calibration for Aligned Large Language Models: A Calibration-Aware Fine-Tuning Approach

ICML 2025poster

One of the key technologies for the success of Large Language Models (LLMs) is preference alignment. However, a notable side effect of preference alignment is poor calibration: while the pre-trained models are typically well-calibrated, LLMs tend to become poorly calibrated after alignment with huma…

Cited by 0SourcePDFScholar