arXiv · 2401.17400
Efficient Learning of Convolution Weights as Gaussian Mixture Model Posteriors
Abstract
In this paper, we showed that the feature map of a convolution layer is equivalent to the unnormalized log posterior of a special kind of Gaussian mixture for image modeling. Then we expanded the model to drive diverse features and proposed a corresponding EM algorithm to learn the model. Learning convolution weights using this approach is efficient, guaranteed to converge, and does not need supervised information. Code is available at: https://github.com/LifanLiang/CALM.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Lifan Liang. 2024-07-01. Efficient Learning of Convolution Weights as Gaussian Mixture Model Posteriors. https://arxiv.org/abs/2401.17400
Cite the original work for its findings. Save a collection to share your selection of sources.