Qwen3-0.6B-KTO

<a href="https://aiplans.org" target="_blank" style="margin: 2px;"> <img alt="AIPlans" src="./logos/AI-Plans Logo (Icon and Text).svg" style="display: inline-block; vertical-align: middle;"/> </a>

Qwen3-0.6B-KTO

Model Card for Model ID

This model is a fine-tuned variant of Qwen/Qwen3-0.6B, trained using KTO (Kahneman-Tversky Optimization) on the nvidia/HelpSteer2 dataset as part of the AIPlans Model Diffing Project.

Model Details

Model Description

This model is a 0.6B parameter language model based on Qwen3-0.6B and fine-tuned using KTO for preference optimization. The goal of the fine-tuning was to improve helpfulness/harmlessness behavior as measured by the HelpSteer2 dataset, while also enabling controlled model diffing experiments as part of the AIPlans research workflow.

Special care was taken to reduce GPU memory usage during KTO training, including gradient checkpointing, selective layer freezing, and reduced forward-pass caching.

Developed by: AIPlans

Funded by : AIPlans

Shared by: AIPlans

Model type: Causal decoder-only Transformer (LLM)

Languages: English

License: MIT (inherits from base model and dataset licensing)

Fine-tuned from: Qwen/Qwen3-0.6B

Training Method: Kahneman-Tversky Optimization (KTO)

Intended Use: Research on model diffing, preference fine-tuning, evaluation of lightweight LLM behavior changes

Model Sources

<!-- Provide the basic links for the model. -->

• Repository: https://github.com/AI-Plans/Model-Diffing/tree/main/KTO-Trainer
• KTO Paper: https://arxiv.org/abs/2402.01306

Training Details

Training Data

Used HelpSteer2 dataset by Nvidia. To convert the dataset into preference dataset we used threshold of 3 on helpfulness. Anything below three belongs to rejected class. Anything above three belongs to accepted class. Data points that were equal 3 were discarded

For more information refer to the github script

Evaluation

Below is a comparison between the base Qwen3-0.6B model and our KTO-trained version (trained using HelpSteer2 preference data).

📊 Benchmark Comparison

📌 Summary

The KTO-trained model shows:

• Improvements on ARC-Easy, HellaSwag, and TruthfulQA (mc2)
• Minor regressions on ARC-Challenge and WinoGrande (within standard error)
• Overall better truthfulness and instruction-following without compromising reasoning ability

These results indicate that preference optimization using HelpSteer2 improves alignment and factual helpfulness while maintaining core model capabilities.

Model Card Authors

Jithesh Pavan D Souza - AIPlans Research Intern

Model Card Contact

Jithesh - [email protected]