arXiv:2512.00792cs.LG2025-12

通过低秩分解量化ViT模型真实有效维度,发现关键压缩区间。

Estimating the Effective Rank of Vision Transformers via Low-Rank Factorization

论文配图:Estimating the Effective Rank of Vision Transformers via Low-Rank Factorization
图 1 · 摘自论文原文
  • 用教师-学生蒸馏法,分阶压缩权重并测性能变化。
  • 在CIFAR-100上有效秩区间为[16,34],32秩时达教师94.7%精度。
  • 提出‘有效拐点’概念,定位性能提升最密集的压缩点。

深度网络高度过参数化,但其学习表征常具有低秩结构。本文提出一种框架,将模型表征视为对全容量低秩子空间的投影,通过训练全秩教师模型,对其权重在多秩下进行分解,并用蒸馏训练对应的学生模型,以衡量性能随秩的变化。定义有效秩为一个区间:使学生模型达到教师准确率85%-95%的最小连续秩范围。为稳定估计,采用单调PCHIP插值拟合准确率-秩曲线,识别归一化曲线交叉点。同时定义有效拐点为平滑后准确率曲线与终点弦之间垂直距离最大的秩,作为边际收益集中点的内在指标。在基于CIFAR-100微调的ViT-B/32上(单次种子),分解线性模块并蒸馏训练后,有效秩区间约为[16, 34],有效拐点约r* ~ 31。在秩32时,学生模型取得69.46%的top-1准确率,对比教师的73.35%(约94.7%基线),实现显著参数压缩。本框架可推广至多种架构与数据集,提供实用工具以刻画深层模型的内在维度。

原文摘要 · Abstract (English)

Deep networks are heavily over-parameterized, yet their learned representations often admit low-rank structure. We introduce a framework for estimating a model's intrinsic dimensionality by treating learned representations as projections onto a low-rank subspace of the model's full capacity. Our approach: train a full-rank teacher, factorize its weights at multiple ranks, and train each factorized student via distillation to measure performance as a function of rank. We define effective rank as a region, not a point: the smallest contiguous set of ranks for which the student reaches 85-95% of teacher accuracy. To stabilize estimates, we fit accuracy vs. rank with a monotone PCHIP interpolant and identify crossings of the normalized curve. We also define the effective knee as the rank maximizing perpendicular distance between the smoothed accuracy curve and its endpoint secant; an intrinsic indicator of where marginal gains concentrate. On ViT-B/32 fine-tuned on CIFAR-100 (one seed, due to compute constraints), factorizing linear blocks and training with distillation yields an effective-rank region of approximately [16, 34] and an effective knee at r* ~ 31. At rank 32, the student attains 69.46% top-1 accuracy vs. 73.35% for the teacher (~94.7% of baseline) while achieving substantial parameter compression. We provide a framework to estimate effective-rank regions and knees across architectures and datasets, offering a practical tool for characterizing the intrinsic dimensionality of deep models.

模型压缩有效秩ViT低秩分解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。