arXiv reaDer
WavePaint: 自己監視型修復のためのリソース効率の高いトークンミキサー
WavePaint: Resource-efficient Token-mixer for Self-supervised Inpainting
画像修復は、画像内の欠落領域を合成することを指し、遮蔽された領域または劣化した領域を復元するのに役立ち、自己監視の前段階タスクとしても機能します。画像修復のための現在の最先端のモデルは、敵対的または拡散設定でトレーニングされたトランスフォーマーまたは CNN バックボーンに基づいているため、計算量が多くなります。この論文は、計算効率の高い WaveMix ベースの完全畳み込みアーキテクチャである WavePaint を使用することで、ビジョン トランスフォーマーから分岐しています。畳み込み層とともに空間および多重解像度のトークン混合に 2D 離散ウェーブレット変換 (DWT) を使用します。提案されたモデルは、画像修復の現在の最先端のモデルよりも再構成品質の点で優れており、使用するパラメーター数も半分以下で、トレーニングと評価の時間も大幅に短縮されます。私たちのモデルは、敵対的にトレーニング可能な識別器を使用せずに、CelebA-HQ データセットの現在の GAN ベースのアーキテクチャをも上回るパフォーマンスを発揮します。私たちの研究は、自然画像事前分布に基づいてモデル化されたニューラル アーキテクチャが、トランスフォーマーに匹敵する一般化を達成するために必要なパラメータと計算が少ないことを示唆しています。
Image inpainting, which refers to the synthesis of missing regions in an image, can help restore occluded or degraded areas and also serve as a precursor task for self-supervision. The current state-of-the-art models for image inpainting are computationally heavy as they are based on transformer or CNN backbones that are trained in adversarial or diffusion settings. This paper diverges from vision transformers by using a computationally-efficient WaveMix-based fully convolutional architecture -- WavePaint. It uses a 2D-discrete wavelet transform (DWT) for spatial and multi-resolution token-mixing along with convolutional layers. The proposed model outperforms the current state-of-the-art models for image inpainting on reconstruction quality while also using less than half the parameter count and considerably lower training and evaluation times. Our model even outperforms current GAN-based architectures in CelebA-HQ dataset without using an adversarially trainable discriminator. Our work suggests that neural architectures that are modeled after natural image priors require fewer parameters and computations to achieve generalization comparable to transformers.
updated: Sat Jul 01 2023 18:41:34 GMT+0000 (UTC)
published: Sat Jul 01 2023 18:41:34 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト