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.
☆56Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torch_kmeans
- ☆40Updated last year
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆172Updated last year
- Transformers w/o Attention, based fully on MLPs☆87Updated 7 months ago
- ☆41Updated last year
- The official github repo for "Test-Time Training with Masked Autoencoders"☆75Updated 11 months ago
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆60Updated 6 months ago
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆55Updated 11 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- [ICLR 2024] Improving Convergence and Generalization Using Parameter Symmetries☆28Updated 5 months ago
- PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"☆48Updated last year
- Decoupled Kullback-Leibler Divergence Loss (DKL), NeurIPS 2024☆32Updated 2 weeks ago
- List of papers that combine self-supervision and continual learning☆59Updated this week
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆138Updated 2 years ago
- Visualizing representations with diffusion based conditional generative model.☆83Updated last year
- [NeurIPS21] TTT++: When Does Self-supervised Test-time Training Fail or Thrive?☆58Updated 2 years ago
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆76Updated last year
- Denoising Masked Autoencoders Help Robust Classification.☆60Updated last year
- This is a pytorch implementation of k-means clustering algorithm☆285Updated 8 months ago
- [ICLR 2023] Masked Frequency Modeling for Self-Supervised Visual Pre-Training☆68Updated last year
- [ICLR 2024 Spotlight] Neuron Activation Coverage: Rethinking Out-of-distribution Detection and Generalization☆25Updated 2 weeks ago
- C-Mixup for NeurIPS 2022☆66Updated 10 months ago
- ☆63Updated 2 years ago
- Pytorch implementation for ICLR 2021 paper - MiCE: Mixture of Contrastive Experts for Unsupervised Image Clustering☆46Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning☆36Updated 6 months ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆67Updated last year
- Official implementation of the paper "Masked Autoencoders are Efficient Class Incremental Learners"☆36Updated 5 months ago
- State Space Models☆61Updated 6 months ago
- Towards Unified and Effective Domain Generalization☆29Updated 11 months ago
- ☆22Updated 3 years ago