Code for "Keypoint Aware Masked Image Modelling"

Custom groovy scripts for QuaPath

Self-supervised learning for image segmentation in PyTorch

Learning Cortical Anomaly through Masked Encoding for Unsupervised Heterogeneity Mapping.

Medical supervised masked autoencoders: Crafting a better masking strategy and efficient fine-tuning schedule for medical image classification

Self-Supervised Representation Learning of Semiconductor Wafer Maps using PyTorch

code for "AdPE: Adversarial Positional Embeddings for Pretraining Vision Transformers via MAE+"

[WACV'25] Official implementation of "PK-YOLO: Pretrained Knowledge Guided YOLO for Brain Tumor Detection in Multiplane MRI Slices".

Masked Autoencoder Pretraining on 3D Brain MRI

Pytorch reimplementation of "A Unified View of Masked Image Modeling".

[MICCAI 2024] HySparK: Hybrid Sparse Masking for Large Scale Medical Image Pre-Training

PyTorch implementation for "Training and Inference on Any-Order Autoregressive Models the Right Way", NeurIPS 2022 Oral, TPM 2023 Best Paper Honorable Mention

Pytorch implementation of an energy transformer - an energy-based reccurrent variant of the transformer.

[ECCV 2024] Official PyTorch implementation of LUT "Learning with Unmasked Tokens Drives Stronger Vision Learners"

Official implementation of Matrix Variational Masked Autoencoder (M-MAE) for ICML paper "Information Flow in Self-Supervised Learning" (https://arxiv.org/abs/2309.17281)

Code of CropMix: Sampling a Rich Input Distribution via Multi-Scale Cropping

[NeurIPS 2023] Masked Image Residual Learning for Scaling Deeper Vision Transformers

MambaMIM: Pre-training Mamba with State Space Token-interpolation

Code to reproduce experiments from the paper "Continual Pre-Training Mitigates Forgetting in Language and Vision" https://arxiv.org/abs/2205.09357