香港中文大學(xué)開(kāi)源 自監(jiān)督表征學(xué)習(xí)代碼庫(kù)

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