香港中文大学开源 自监督表征学习代码库

近日，香港中文大學多媒体实验室（MMLab）和南洋理工大学的研究者最近开源了一套统一的自监督学习代码库OpenSelfSup。OpenSelfSup使用PyTorch实现，支持基于分类、重建、聚类、memory bank、contrastive learning的多种自监督学习框架，目前收录了NPID、MoCo、SimCLR等一系列表现较好的自监督表征学习方法，后续还将陆续跟进学术界最新算法。在这个框架下，每个算法被拆解为backbone、neck、head等多个可独立设计的模块，每个模块均提供多个可选方案，开发者也可以自行设计各个模块。