arXiv論文メモ
新着一覧
cs.LG / cs.CV · 査読状況未確認

画像モデルの積み重ねで精度と確率の較正を改善する方法

CORE-STACK+: Meta-Learning for Deep Stacked Generalization

Noor Islam S. Mohammad

この論文をやさしく読む

ひとことで言うと

複数の画像モデルの予測を組み合わせる際、重複と確率のずれを同時に抑える方法。

何に役立つ?

画像モデルの精度や較正、推論量の改善に役立つ可能性がある。要旨の実績は六つのベンチマークでの評価である。

この研究の面白いところ

モデル間の非線形な重複を除き、標本ごとの重み付けとベイズ型の混合を組み合わせている。

どこまで分かった?

数値結果は記載されたベンチマークと比較条件でのもの。要旨はあらゆる画像課題への一般化を示していない。

v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。

アブストラクトの日本語訳

CNN、ViT、両者を組み合わせた画像モデルなどを積み重ねることは、精度、確率の較正、頑健性を高める一般的な方法だが、二つの関連した問題が効果を制限する。予測同士の多重共線性はメタ学習器のグラム行列の条件を悪くし、重みのばらつきを増やして薄い多様体上の不安定な解を生む。また、単純な線形の積み重ねは各モデルの較正不良を重ねるため、モデルを追加すると期待較正誤差(ECE)が悪化しうる。リッジ正則化、貪欲な選択、モデルスープ、SWAGといった従来の対策は、これらの問題の一方にしか対処せず、異種モデルの予測群で条件付けと較正を同時に狙うものはない。 著者らは四つの要素からなる前処理パイプラインCORE-STACK+を提案する。第一にCentered Kernel Alignment(CKA)を使い、ピアソン相関では見えないモデル間の非線形な依存関係を除くカーネル型の冗長性フィルター。第二に、アンサンブルの統計量に対して標本ごとの注意を学ぶ、パラメータ数1万5千未満の微分可能なメタ特徴ゲート。第三に、Marchenko–Pasturの信号・雑音分解から導いた、スペクトルに適応するリッジ罰則λ*=λmax(Ĉ)/SNR(Ĉ)で、入れ子の交差検証を不要にする。第四に、逆RMSEの経験則に代わるラプラス近似のベイズ型混合器である。予測空間の冗長性とメタ学習器の容量を同時に考慮するPACベイズの超過リスク界を初めて証明したと述べる。六つのベンチマークで、ImageNet-1Kのtop-1精度は1.8%向上、ImageNet-CのmCEは4.2低下、ADE20KのmIoUは0.9向上、COCOのAPは1.3向上した。残すモデル数は35~57%減り、推論FLOPsは最大41%減った。事後の温度スケーリングなしで、深層アンサンブルに比べECEは2.1倍改善した。

v1の要旨から自動生成。本文の精読・人による確認は未実施。

初稿
2026-09-22(UTC)
最新改訂
2026-09-22 · v1
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

Stacking heterogeneous vision backbones (CNNs, ViTs, and hybrids) is the de facto recipe for accuracy, calibration, and robustness, yet two coupled pathologies limit its returns. Prediction-space multicollinearity ill-conditions the meta-learner's Gram matrix, inflating weight variance and producing brittle solutions on a thin manifold. Calibration collapse compounds constituent miscalibration through naive linear stacking, so adding more models can hurt expected calibration error (ECE). Existing remedies, ridge regularization, greedy selection, model soups, and SWAG address at most one of these issues, and none jointly target conditioning and calibration in heterogeneous prediction pools. We introduce CORE-STACK+, a preconditioning pipeline with four components: (i) a kernelized redundancy filter that removes non-linear inter-model dependencies invisible to Pearson correlation, using Centered Kernel Alignment (CKA) [23]; (ii) a $<15$K-parameter differentiable meta-feature gate that learns per-sample attention over ensemble statistics; (iii) a spectrum-adaptive Ridge penalty $lambda^{star}=lmax(Chat)/SNR(Chat)$ derived from a Marchenko-Pastur signal-noise decomposition, eliminating nested cross-validation; and (iv) a Laplace-approximate Bayesian blender replacing inverse-RMSE heuristics. We prove a PAC-Bayes excess-risk bound that, for the first time, jointly accounts for prediction-space redundancy and meta-learner capacity. Across six benchmarks, CORE-STACK+ delivers $+1.8\%$ top-1 on ImageNet-1K, $-4.2$ mCE on ImageNet-C, $+0.9$ mIoU on ADE20K, and $+1.3$ AP on COCO, while reducing retained models by 35-57% and inference FLOPs by up to $41%$. ECE improves $2.1\times$ over deep ensembles without post hoc temperature scaling.

arXiv ID: 2609.26905 / 要約の誤りについて