QuantLRM: Quantization of Large Reasoning Models via Fine-Tuning Signals

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Nan, Kwek, Eugene, Zhang, Yusen, Pan, Muyu, Wang, Suhang, Mitra, Prasenjit, Zhang, Rui
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918320488316928
author Zhang, Nan
Kwek, Eugene
Zhang, Yusen
Pan, Muyu
Wang, Suhang
Mitra, Prasenjit
Zhang, Rui
author_facet Zhang, Nan
Kwek, Eugene
Zhang, Yusen
Pan, Muyu
Wang, Suhang
Mitra, Prasenjit
Zhang, Rui
contents Weight-only quantization is important for compressing Large Language Models (LLMs). Inspired by the spirit of classical magnitude pruning, we study whether the magnitude of weight updates during reasoning-incentivized fine-tuning can provide valuable signals for quantizing Large Reasoning Models (LRMs). We hypothesize that the smallest and largest weight updates during fine-tuning are more important than those of intermediate magnitude, a phenomenon we term "protecting both ends". Upon hypothesis validation, we introduce QuantLRM, which stands for weight quantization of LRMs via fine-tuning signals. We fit simple restricted quadratic functions on weight updates to protect both ends. By multiplying the average quadratic values with the count of zero weight updates of channels, we compute channel importance that is more effective than using activation or second-order information. We run QuantLRM to quantize various fine-tuned models (including supervised, direct preference optimization, and reinforcement learning fine-tuning) over four reasoning benchmarks (AIME-120, FOLIO, temporal sequences, and GPQA-Diamond) and empirically find that QuantLRM delivers a consistent improvement for LRMs quantization, with an average improvement of 6.55% on a reinforcement learning fine-tuned model. Also supporting non-fine-tuned LRMs, QuantLRM gathers effective signals via pseudo-fine-tuning, which greatly enhances its applicability.
format Preprint
id arxiv_https___arxiv_org_abs_2602_02581
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle QuantLRM: Quantization of Large Reasoning Models via Fine-Tuning Signals
Zhang, Nan
Kwek, Eugene
Zhang, Yusen
Pan, Muyu
Wang, Suhang
Mitra, Prasenjit
Zhang, Rui
Machine Learning
Artificial Intelligence
Weight-only quantization is important for compressing Large Language Models (LLMs). Inspired by the spirit of classical magnitude pruning, we study whether the magnitude of weight updates during reasoning-incentivized fine-tuning can provide valuable signals for quantizing Large Reasoning Models (LRMs). We hypothesize that the smallest and largest weight updates during fine-tuning are more important than those of intermediate magnitude, a phenomenon we term "protecting both ends". Upon hypothesis validation, we introduce QuantLRM, which stands for weight quantization of LRMs via fine-tuning signals. We fit simple restricted quadratic functions on weight updates to protect both ends. By multiplying the average quadratic values with the count of zero weight updates of channels, we compute channel importance that is more effective than using activation or second-order information. We run QuantLRM to quantize various fine-tuned models (including supervised, direct preference optimization, and reinforcement learning fine-tuning) over four reasoning benchmarks (AIME-120, FOLIO, temporal sequences, and GPQA-Diamond) and empirically find that QuantLRM delivers a consistent improvement for LRMs quantization, with an average improvement of 6.55% on a reinforcement learning fine-tuned model. Also supporting non-fine-tuned LRMs, QuantLRM gathers effective signals via pseudo-fine-tuning, which greatly enhances its applicability.
title QuantLRM: Quantization of Large Reasoning Models via Fine-Tuning Signals
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2602.02581