← Search

Yi-Kai Zhang

8 accepted papers

2026

$V_0$: A Generalist Value Model for Any Policy at State Zero

ICML 2026poster

Traditional value models $V^{\pi}$ in LLM reinforcement learning face a coupling dilemma: they require synchronous training alongside the updating policy $\pi$, causing inefficiency and overfitting. In this paper, we propose $V_0$, a generalist value model that decouples value estimation from specif…

Cited by 0SourceScholar
2025

Let the LLM Stick to Its Strengths: Learning to Route Economical LLM

NeurIPS 2025poster

Recently, test-time scaling of Large Language Models (LLMs) has emerged as a practical alternative to parameter and data scaling. Reasoning tasks often require large-scale, RLVR-based LLMs, while more economical LLMs can handle simpler tasks. Routing an LLM tailored to *suitability* (*i.e.*, capabil…

Cited by 0SourceScholar
2025

ZooProbe: A Data Engine for Evaluating, Exploring, and Evolving Large-scale Training Data for Multimodal LLMs

ICLR 2025poster

Multimodal Large Language Models (MLLMs) are thriving through continuous fine-tuning by LLMs. Driven by the law that "scale is everything", MLLMs expand their training sets during version iterations. In this paper, we propose a large-scale training data engine built around an evaluating-exploring-ev…

Cited by 0SourcePDFScholar
2024

Wings: Learning Multimodal LLMs without Text-only Forgetting

NeurIPS 2024poster

Multimodal large language models (MLLMs), initiated with a trained LLM, first align images with text and then fine-tune on multimodal mixed inputs. However, during the continued training, the MLLM catastrophically forgets the text-only instructions that the initial LLM masters. In this paper, we pre…

Cited by 11SourcePDFScholar
2023

Few-Shot Class-Incremental Learning via Training-Free Prototype Calibration

NeurIPS 2023poster

Real-world scenarios are usually accompanied by continuously appearing classes with scare labeled samples, which require the machine learning model to incrementally learn new classes and maintain the knowledge of base classes. In this Few-Shot Class-Incremental Learning (FSCIL) scenario, existing me…

2023

Learning Debiased Representations via Conditional Attribute Interpolation

CVPR 2023poster

An image is usually described by more than one attribute like "shape" and "color". When a dataset is biased, i.e., most samples have attributes spuriously correlated with the target label, a Deep Neural Network (DNN) is prone to make predictions by the "unintended" attribute, especially if it is eas…

2023

Model Spider: Learning to Rank Pre-Trained Models Efficiently

NeurIPS 2023spotlight

Figuring out which Pre-Trained Model (PTM) from a model zoo fits the target task is essential to take advantage of plentiful model resources. With the availability of numerous heterogeneous PTMs from diverse fields, efficiently selecting the most suitable one is challenging due to the time-consuming…