Module providing Consciousness Exploration Tools for PyTorch.
API documentation of consciousnet.models¶
consciousnet.models¶
Common architectures.
consciousnet.models.brainnetcnn¶
BrainNetCNNs are convolutional neural networks for connectomes.
consciousnet.models.distributions¶
Common distributions.
consciousnet.models.tdvae¶
Definition of the Temporal Difference Variational Auto-Encoder (TD-VAE) model.
consciousnet.models.barlowtwins¶
Definition of the Self-Supervised Learning via Redundancy Reduction (Barlow Twins) model.
consciousnet.models.gmvae¶
Definition of the Gaussian Mixture Variational Auto-Encoder (GMVAE) models.
Two implementations are proposed:
VAEGMP is an adaptation of VAE to make use of a Gaussian Mixture prior, instead of a standard Normal distribution.
GMVAE is an attempt to replicate the work described in [1] and [2].
[1] Gaussian Mixture VAE: Lessons in Variational Inference, Generative Models, and Deep Nets: http://ruishu.io/2016/12/25/gmvae [2] Deep Unsupervised Clustering with Gaussian Mixture Variational Autoencoders Nat Dilokthanakul, arXiv 2017. Code: https://github.com/jariasf/GMVAE Code: https://github.com/mazrk7/gmvae
Follow us