DRWKV: Focusing on Object Edges for Low-Light Image Enhancement

基本情報

• arXiv ID: 2507.18594 (https://arxiv.org/abs/2507.18594)
• 著者: Xuecheng Bai, Yuxiang Wang, Boyu Hu他
• 所属: Shenyang Ligong University, The University of Sydney他
• 投稿日: 2025年07月26日
• カテゴリ: cs.CV, eess.IV

簡単に説明すると

この論文は、低照度画像の強調において物体のエッジの連続性と細部の保存に焦点を当てた新手法「DRWKV（Detailed Receptance Weighted Key Value）」を提案しています。
極端な照明劣化の条件下でもエッジの忠実性を保つため、Global Edge Retinex（GER）理論を導入します。
これにより、照明とエッジ構造を効果的に分離します。
また、Evolving WKV Attentionという螺旋スキャンメカニズムを採用し、空間的なエッジの連続性を捉えます。
5つのベンチマークデータセットでの実験により、低い計算コストでPSNR、SSIM、NIQEで優れた性能を達成しています。

1. 研究概要

1.1 背景と動機

低照度画像の強調（LLIE）技術は、低コストで暗い画像を明るくする能動的な手法として注目されています。
従来の手法では、グローバルな照明のみに焦点を当てると背景ノイズの増幅とエッジの歪みが発生します。
一方、輪郭のみに焦点を当てると、急激な遷移と全体的な露出不足を引き起こします。
既存のVRWKV（Visual RWKV）モデルは、低照度環境での物体エッジの一貫性モデリングに限界があります。
エッジのぼやけや照明変動との混在、明暗領域でのノイズとエッジの誤判定などの問題を抱えていました。

1.2 主要な貢献

• Global Edge Retinex（GER）理論を初めてVRWKVモデルと統合し、低照度画像のエッジを効果的に強調
• 内側から外側へエッジの詳細を抽出するEvolving Scanning（ES）メカニズムの設計とWKV Attentionへの適用
• スペクトル特徴を調整し、色域を豊かにするBilateral Spectrum Aligner（Bi-SAB）の導入
• エッジの連続性と詳細の忠実性において優れた性能を実証

2. 提案手法

2.1 手法の概要

DRWKVは階層的な最適化戦略を採用し、まずLight Preprocessingで輝度復元画像を取得します。
その後、Evolving Scanningを適用したRWKV構造によるDeep Detail Miningを行います。
エンコーダーはEvolving Scanningとウェーブレット変換ダウンサンプリングを組み合わせてエッジ勾配特徴を抽出します。
デコーダーはBi-SABを介してサンプリング情報を統合し、微視的テクスチャから巨視的輪郭まで段階的に強調します。

2.2 技術的詳細

Global Edge Retinex（GER）理論：
従来のRetinex理論を拡張し、以下の定式化を提案しています。

I = (R + α·E) ⊙ L + β·N + γ·S

ここで、Eはエッジ特徴項、Sはアーティファクト項、Nは空間的に不均一なノイズを表します。

Evolving Scanning（ES）メカニズム：
アルキメデスの螺旋方程式に基づき、エッジマニフォールドの幾何的な連続性を時間的連続性に変換します。

P(θ) = (r(θ)cosθ, r(θ)sinθ)

単一の螺旋を4方向螺旋システムに拡張し、時計回りと反時計回りの両方向をサポートします。

Bilateral Spectrum Aligner（Bi-SAB）：
CrossAttentionを通じて輝度と色分岐間の特徴相互作用を実現します。
Spectral Alignment Enhancer（SAE）で画像の明るさを向上させ、ノイズを低減します。
Scharrオペレータでエッジの詳細を抽出します。

MS²-Loss：
構造、エッジ、照明、アーティファクト、重みの5つの主要属性に焦点を当てた新しい損失関数を定義します。

2.3 新規性

• エッジの忠実性を保つためのGER理論の提案
• 低照度エッジの空間トポロジーに適応するEvolving Scanning技術
• 輝度と色度の特徴を整合させるBi-SABの設計
• 包括的なMS²-Loss関数の導入

3. 実験結果

3.1 実験設定

データセットには以下を使用しました。

• LSRW-Huawei：2,450組の訓練画像と30組のテスト画像
• LOLv2（Real/Synthetic）：実環境と合成環境の低照度データセット
• SDSD（indoor/outdoor）：屋内外の低照度動画データセット

評価指標： PSNR、SSIM、NIQE

実装詳細： NVIDIA A800-80GB、エポック数500、入力サイズ128×128、Adam optimizer使用。

3.2 主要な結果

LSRW-Huaweiデータセットでの結果は以下のとおりです。

• PSNR: 22.34（最高値）
• SSIM: 0.703
• NIQE: 2.944（最低値＝最良）

LOLv2 Realデータセットでの結果は以下のとおりです。

• PSNR: 24.12（最高値、従来手法より約2dB向上）
• SSIM: 0.832
• NIQE: 3.926（最低値）

計算効率の結果は以下のとおりです。

• GFLOPs: 1.67（RetinexNetの587.47と比較して350分の1以下）

3.3 既存手法との比較

DRWKVは、RetinexNet、KinD、Zero-DCE、LLFormer、RetinexMamba、MambaLLIEなどの既存手法と比較して、全ての評価指標で優れた性能を示しました。
特に、計算コストが約350分の1に削減されたにもかかわらず、優れた画質改善を達成しています。
拡張実験として、低照度マルチオブジェクトトラッキングタスクでも評価し、ByteTrackで5.189%のMOTA向上を実現しました。

4. 実用性評価

4.1 実装の容易性

モデルアーキテクチャは比較的シンプルで、既存のディープラーニングフレームワークで実装可能です。
論文中でコードの公開について言及はありませんが、詳細な実装説明が提供されています。

4.2 計算効率

1.67 GFLOPsという非常に低い計算コストで動作し、リアルタイム処理に適しています。
パラメータ数も8.28Mと比較的少なく、モバイルデバイスへの展開も期待できます。

4.3 応用可能性

• 監視カメラシステムでの夜間映像強調
• 自動運転車の夜間視認性の向上
• スマートフォンでの低照度撮影の補正
• 医療画像処理での暗部詳細強調
• ドローンによる夜間撮影の品質向上

5. まとめと所感

5.1 論文の意義

この研究は、低照度画像の強調において長年の課題であったエッジの保存問題に対する革新的な解決策を提供しています。
特に、GER理論による照明とエッジの分離は理論的に重要な貢献です。
Evolving Scanningによる空間的連続性の保持は実践的に重要な貢献です。
計算効率と性能のバランスが優れており、実用化への道筋が明確である点も評価できます。

5.2 今後の展望

• モバイルデバイス向けの更なる軽量化と最適化
• 動画処理への拡張と時間的一貫性の確保
• 他の画像復元タスク（デノイジング、超解像など）への応用
• エンドツーエンドの学習フレームワークへの統合
• リアルタイムビデオストリーミングへの適用

DRWKVは、低照度画像の強調分野において重要なブレークスルーとなる可能性があります。
特にエッジ保存型の画像処理手法として今後の発展が期待されます。