香港中文大學開源 自監督表征學習代碼庫

近日，香港中文大學多媒體實驗室（MMLab）和南洋理工大學的研究者最近開源了一套統一的自監督學習代碼庫OpenSelfSup。OpenSelfSup使用PyTorch實現，支持基于分類、重建、聚類、memory bank、contrastive learning的多種自監督學習框架，目前收錄了NPID、MoCo、SimCLR等一系列表現較好的自監督表征學習方法，后續還將陸續跟進學術界最新算法。在這個框架下，每個算法被拆解為backbone、neck、head等多個可獨立設計的模塊，每個模塊均提供多個可選方案，開發者也可以自行設計各個模塊。