deepinv/deepinv

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

解決する課題

DeepInverseは、イメージング逆問題の解決という課題に取り組んでいます。さまざまなイメージング領域におけるディープラーニング研究を加速させるためのモジュール式フレームワークを提供し、再現性を高め、新規実務者の参入障壁を下げます。

仕組み

このライブラリはPyTorchに基づいて構築されており、以下の主要コンポーネントを提供します:

  • 事前に定義されたイメージングオペレーターの広範なフレームワーク。
  • 学習済み再構成モデルやデノイザーを含む、最先端のディープニューラルネットワーク。
  • プラグアンドプレイ形式の復元、最適化、および展開型(unfolded)アーキテクチャのためのフレームワーク。
  • 逆問題に特化したトレーニング損失関数。
  • 不確実性定量化のためのサンプリングアルゴリズムと拡散モデル。
  • 逆問題のためのデータセット構築に特化したフレームワーク。

対象者

イメージング領域で研究や実務に従事しており、逆問題のためのディープラーニングアルゴリズムを実装・テストするための標準化されたモジュール方式を必要とする研究者や実務者向けに設計されています。

ハイライト

  • PyTorchエコシステムの公式の一部
  • プラグアンドプレイ形式の復元および展開型アーキテクチャをサポート
  • 不確実性定量化のための拡散モデルを搭載
  • 特化型データセット構築のためのフレームワークを提供