Efficiently Learning at Test-Time: Active Fine-Tuning of LLMs

Applying Active Learning to fine-tune LLMs

Overview

Large language models (LLMs) are typically trained once and then deployed without adapting to new information or user-specific data. This project investigates how LLMs can efficiently learn during inference by selectively fine-tuning themselves on informative examples encountered at test time.


Project screenshot
Visualization of Active Learning.

Research Contribution

My contributions included developing components of the proposed active fine-tuning framework and conducting experiments to measure the performance.

Publication

This work was published at ICLR 2025.

📄

View publication

Read the paper published at ICLR 2025.

→

Technologies

Results

The project demonstrated that language models can improve their performance through targeted adaptation at test time, reducing the need for expensive full-scale retraining. The proposed active fine-tuning approach enables models to focus computational resources on the most informative examples, improving the efficiency of adaptation.

Project screenshot
Performance improvement of our approach SIFT.
Project screenshot
Retrival example for SIFT and the state of the art method.

← Back to Home