アーカイブ / Computer Vision
画像の細部を再現するINRで、重みの更新方法を変えると何が起きるか
Muon
Muonの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1 : Rank preservation overcomes the low-frequency bias of ReLU MLPs. Vanilla ReLU MLPs lack high-frequency detail (top). Orthogonalized, rank-preserving weight updates enable them to achieve faithful high-frequency reconstructions (middle) by maintaining a stable layer rank throughout optimization (illustrated for hidden layers trained with Adam, which exhibits rank collapse, versus Muon, which preserves stable rank; bottom).
概要
- 画像の座標から色を返すINRで、Muonを使い隠れ層の重み更新を直交に近づける方法を検討した。
- 著者らのKodak画像24枚での比較では、ReLU MLPの平均PSNRがAdamの22.83 dBからMuonの30.04 dBになった。
- 点検写真への利用は応用案であり、細い欠陥の再現、学習時間、点検判断への効果を別途検証する必要がある。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、Muonの仕組み
- 01画素座標とRGB値
- 02MLPの勾配計算
- 03Muonで隠れ層更新
- 04座標からRGB値
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
INR（暗黙的ニューラル表現）は、画像なら位置を入力すると、その位置の色を返す関数として信号を表す方法である。単純な多層パーセプトロン（MLP）では細部を再現しにくいと考えられ、座標の変換や特殊な活性化関数が使われてきた。
一方、表現力の高い構造は、観測の足りない画像復元などで振動状の模様を生む場合があり、調整を要する。著者らは構造の変更に加え、学習中に重みの更新方向が偏ることに着目した。
手法
画像を学習する際は、各画素の座標をMLPに入力し、その画素の赤・緑・青の値を出力できるように重みを調整する。たとえば動物の写真なら、毛並みや輪郭にある座標でも元の色に近づける。著者らは、通常の学習では重みの更新が少数の方向に偏り、細部の学習を妨げ得ると説明する。
偏りを調べる量が安定ランクで、行列の値がどれだけ多くの方向に分散しているかを示す。Muonは隠れ層の重み更新を、方向同士の重なりが小さい「直交」に近い形へ変える。本文はその考え方を特異値分解で示し、Muonでは近似計算を使うと説明する。バイアスなどはAdamで更新し、学習後は座標から色を出力する。
新規性
著者らは、単純な多層パーセプトロン（MLP）が細部を学びにくい理由を、構造だけでなく、学習中に重みの安定ランクが下がる現象から説明する。座標の変換や活性化関数の変更も、更新に使える方向を増やす働きとして捉えている。
その見方に基づき、隠れ層の重み更新を直交に近づけるMuonをINRに適用した。同じ構造をAdamで学習した場合と比べ、複数の信号表現で再現精度が変わることを示した。
従来手法との違い
比較の中心は、同じINR構造をAdamまたはMuonで学習した場合である。Kodak画像の表には、単純なReLU MLPに加え、座標を周波数成分に変換するReLU FFNや、異なる活性化関数を使う構造が載っている。各構造について最適化手法ごとの画質を比べられる。
Muonを使う場合、直交化するのは隠れ層の重み更新で、バイアスや入力の符号化に関わるパラメータなどにはAdamを残す。表は再現画質を示すが、学習時間や計算量の差はそこからは判断できない。
実験結果
著者らは768×512画素のKodak画像24枚を用い、画像ごとの再現画質を比較した。ReLU MLPの平均PSNRはAdamで22.83±2.85 dB、Muonで30.04±3.59 dBだった。PSNRは画素値に基づく画質指標で、高いほど元画像に近い。同じ比較で、構造の類似度を測るSSIMは0.546±0.155から0.813±0.074になった。
別の6秒のバッハの音声再現では、ReLU FFNのSNRがAdamの8.23 dB、Muonの19.71 dBと報告されている。SNRは信号と再現誤差の比を表す。確認できた本文の範囲では、付録に記された学習率などの設定全体は照合できない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：設備点検写真の細い傷や文字が、座標から色を返すINRでどの程度残るか調べる。入力には利用許可のある写真を使い、各画素の座標を学習入力、RGB値を正解ラベルにする。写真以外のセンサーは想定しない。
写真ごとに同じMLP構造をAdam版とMuon版で学習し、得られた再現画像を原画像と並べて確認する。現場写真に合わせ、解像度、座標の表し方、学習設定を調整する必要がある。傷の検出や点検判断が改善するかは、この研究結果だけでは分からない。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：利用許可のある点検画像を少数集め、各画素の座標を入力、RGB値を正解ラベルにして画像ごとに学習する。追加センサーは使わない。同一構造のReLU MLPでAdam版とMuon版を作り、前処理、学習回数、評価対象をそろえる。
原画像に対するPSNRとSSIMを測り、傷の位置と形は担当者が再現画像と見比べる。現場画像の解像度に合わせて座標の設定を調整し、学習時間も記録する。画質指標が上がっても点検時の見落としが減るとは限らないため、業務上の判定精度は別に測る。
限界と課題
著者らは、6秒の音声では単純なReLU MLPが最適化手法によらず全体を再現できず、短縮した200ミリ秒の区間ではMuonによる学習が成功したと報告する。また、小規模な形状表現では、周期的な活性化関数を使う一部のINRが収束しなかったという。
編集上の検証事項は、対象画像の細い欠陥が保たれるか、初期値を変えても結果が安定するか、学習時間と計算量が運用に収まるかである。形状の表ではMuonが全指標で優位なわけではなく、画像の結果を別の課題へそのまま当てはめられない。
出典
元のタイトル・要旨を確認する（英語）
Optimizing Rank for High-Fidelity Implicit Neural Representations
Implicit Neural Representations (INRs) based on vanilla Multi-Layer Perceptrons (MLPs) are widely believed to be incapable of representing high-frequency content. This has directed research efforts towards architectural interventions, such as coordinate embeddings or specialized activation functions, to represent high-frequency signals. In this paper, we challenge the notion that the low-frequency bias of vanilla MLPs is an intrinsic, architectural limitation to learn high-frequency content, but instead a symptom of stable rank degradation during training. We empirically demonstrate that regulating the network's rank during training substantially improves the fidelity of the learned signal, rendering even simple MLP architectures expressive. Extensive experiments show that using optimizers like Muon, with high-rank, near-orthogonal updates, consistently enhances INR architectures even beyond simple ReLU MLPs. These substantial improvements hold across a diverse range of domains, including natural and medical images and novel view synthesis, with up to +9 dB PSNR over the same architecture. Code is available at (https://rank-inrs.github.io).
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01