← Search

ZhiHang Lin

9 accepted papers

2026

MoFO: Momentum-Filtered Optimizer for Mitigating Forgetting in LLM Fine-Tuning

ICML 2026poster

Large language models (LLMs) have demonstrated remarkable capabilities across a wide range of tasks. Typically, LLMs are first pre-trained on large corpora and subsequently fine-tuned on task-specific datasets. However, during fine-tuning, LLMs may forget some knowledge acquired in the pre-training …

Cited by 0SourceScholar
2025

Boosting Multimodal Large Language Models with Visual Tokens Withdrawal for Rapid Inference

AAAI 2025technical

Multimodal large language models (MLLMs) demand considerable computations for inference due to the extensive parameters and the additional input tokens needed for visual information representation. Herein, we introduce Visual Tokens Withdrawal (VTW), a plug-and-play module to boost MLLMs for rapid i…

2025

CPPO: Accelerating the Training of Group Relative Policy Optimization-Based Reasoning Models

NeurIPS 2025poster

This paper introduces Completion Pruning Policy Optimization (CPPO) to accelerate the training of reasoning models based on Group Relative Policy Optimization (GRPO). GRPO, while effective, incurs high training costs due to the need to sample multiple completions for each question. Our experiment an…

Cited by 0SourcecodeScholar
2025

Second Language (Arabic) Acquisition of LLMs via Progressive Vocabulary Expansion

ACL 2025long

This paper addresses the critical need for democratizing large language models (LLM) in the Arab world, a region that has seen slower progress in developing models comparable to state-of-the-art offerings like GPT-4 or GPT-3.5, due to a predominant focus on mainstream languages (e.g., English and Ch…

2025

UCL-Bench: A Chinese User-Centric Legal Benchmark for Large Language Models

NAACL 2025findings

Existing legal benchmarks focusing on knowledge and logic effectively evaluate LLMs on various tasks in legal domain. However, few have explored the practical application of LLMs by actual users. To further assess whether LLMs meet the specific needs of legal practitioners in real-world scenarios, w…

2024

AccDiffusion: An Accurate Method for Higher-Resolution Image Generation

ECCV 2024poster

"This paper attempts to address the object repetition issue in patch-wise higher-resolution image generation. We propose AccDiffusion, an accurate method for patch-wise higher-resolution image generation without training. An in-depth analysis in this paper reveals an identical text prompt for differ…

2024

ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models

ICML 2024poster

Reinforcement Learning from Human Feedback (RLHF) is key to aligning Large Language Models (LLMs), typically paired with the Proximal Policy Optimization (PPO) algorithm. While PPO is a powerful method designed for general reinforcement learning tasks, it is overly sophisticated for LLMs, leading to…

2023

SMMix: Self-Motivated Image Mixing for Vision Transformers

ICCV 2023poster

CutMix is a vital augmentation strategy that determines the performance and generalization ability of vision transformers (ViTs). However, the inconsistency between the mixed images and the corresponding labels harms its efficacy. Existing CutMix variants tackle this problem by generating more consi…

Cited by 15PDFcodeScholar
2022

Learning Best Combination for Efficient N:M Sparsity

NeurIPS 2022accept

By forcing N out of M consecutive weights to be non-zero, the recent N:M fine-grained network sparsity has received increasing attention with its two attractive advantages over traditional irregular network sparsity methods: 1) Promising performance at a high sparsity. 2) Significant speedups when p…