jokofa / torch_kmeans
PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
☆63Updated 2 years ago
Alternatives and similar repositories for torch_kmeans:
Users that are interested in torch_kmeans are comparing it to the libraries listed below
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆80Updated last year
- The official github repo for "Test-Time Training with Masked Autoencoders"☆80Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.☆25Updated last year
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆141Updated 3 years ago
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆173Updated last year
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆82Updated last year
- Transformers w/o Attention, based fully on MLPs☆93Updated 11 months ago
- ☆42Updated 2 years ago
- Visualizing representations with diffusion based conditional generative model.☆89Updated last year
- Differentiable Top-k Classification Learning☆80Updated 2 years ago
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆61Updated 10 months ago
- A torch-based implementation of K-Means and K-Means++☆17Updated 4 years ago
- Denoising Masked Autoencoders Help Robust Classification.☆61Updated last year
- State Space Models☆66Updated 10 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆31Updated last year
- ☆147Updated 2 years ago
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆40Updated last year
- List of papers that combine self-supervision and continual learning☆63Updated 4 months ago
- [ICLR 2024] Improving Convergence and Generalization Using Parameter Symmetries☆29Updated 9 months ago
- Official PyTorch code for "Out-of-distribution detection with denoising diffusion models"☆48Updated 9 months ago
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆27Updated 4 months ago
- A conference poster format with structure, content, creation, and presentation recommendations.☆46Updated 3 weeks ago
- This is a PyTorch implementation of “Context AutoEncoder for Self-Supervised Representation Learning"☆102Updated last year
- B-cos Networks: Alignment is All we Need for Interpretability☆107Updated last year
- ☆63Updated 2 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆74Updated last year
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆58Updated last year
- [CVPR 2024] Friendly Sharpness-Aware Minimization☆27Updated 4 months ago