DeepPrism: Channel Convolution for Lightweight Generative Models - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

DeepPrism: Channel Convolution for Lightweight Generative Models

Résumé

In this paper, we introduce DeepPrism, a novel network architecture for generative models, which addresses the redundancy issue of convolutional filters. DeepPrism reaches an unprecedented parameter efficiency improvement of up to 1000 times on generative models. The number of parameters is reduced over conventional CNNs from quadratic to constant dependency with respect to the network’s width. The training / inference time and space are also reduced. The main novelty lies in the geometric property namely the translation equivariance on channels, which gives rise to the convolution structure along the channels, and trivially generalizes to the attention mechanism. Compared with Latent Diffusion Model, DeepPrism produces similar qualitative and quantitative results, but the number of parameters can be reduced at a great scale. Other generative models including autoencoders and single-image diffusion models are also experimented to exhibit the generality of this method on generative models.
Fichier principal
Vignette du fichier
lightweight-generative-model.pdf (3.68 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-04323751 , version 1 (12-01-2024)
hal-04323751 , version 2 (16-01-2024)

Licence

Identifiants

  • HAL Id : hal-04323751 , version 2

Citer

Changqing Fu, Laurent D. Cohen. DeepPrism: Channel Convolution for Lightweight Generative Models. The 5th International Conference on Video, Signal and Image Processing, Nov 2023, Harbin City, virtual, China. ⟨hal-04323751v2⟩
90 Consultations
70 Téléchargements

Partager

More