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.
☆57Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torch_kmeans
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆172Updated last year
- List of papers that combine self-supervision and continual learning☆60Updated last week
- The official github repo for "Test-Time Training with Masked Autoencoders"☆78Updated last year
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆77Updated last year
- A torch-based implementation of K-Means and K-Means++☆18Updated 3 years ago
- ☆40Updated last year
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆75Updated 10 months ago
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆42Updated last year
- Visualizing representations with diffusion based conditional generative model.☆84Updated last year
- B-cos Networks: Alignment is All we Need for Interpretability☆105Updated last year
- [CVPR 2023] This repository includes the official implementation our paper "Masked Autoencoders Enable Efficient Knowledge Distillers"☆99Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- This is a pytorch implementation of k-means clustering algorithm☆286Updated 9 months ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆140Updated 3 years ago
- ☆41Updated last year
- Hyperbolic Image Segmentation, CVPR 2022☆76Updated 2 years ago
- [ICLR 2023] Masked Frequency Modeling for Self-Supervised Visual Pre-Training☆68Updated last year
- Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".☆28Updated last year
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆69Updated last year
- [NeurIPS'23] DropPos: Pre-Training Vision Transformers by Reconstructing Dropped Positions☆60Updated 6 months ago
- This repository contains the implementation for the paper "EMP-SSL: Towards Self-Supervised Learning in One Training Epoch."☆221Updated last year
- [ICML23] On Pitfalls of Test-Time Adaptation☆103Updated 7 months ago
- (Pytorch) Training ResNets on ImageNet-100 data☆52Updated 2 years ago
- Compare neural networks by their feature similarity☆343Updated last year
- Official implementation of AAAI 2023 paper "Parameter-efficient Model Adaptation for Vision Transformers"☆97Updated last year
- Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders☆55Updated last year
- Denoising Masked Autoencoders Help Robust Classification.☆60Updated last year
- [CVPR 2023 Highlight] Masked Image Modeling with Local Multi-Scale Reconstruction☆44Updated last year
- ☆100Updated 8 months ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆75Updated 2 years ago