3月5日消息,Facebook的研究人員推出了一種新的AI模型,該模型可以從網上任何未標記圖像的中學習,這是一項突破,盡管該團隊的研究仍處于早期階段,但他們希望給計算機視覺領域帶來一次“革命”。
該模型被稱為SEER(SElf-SupERvised),被饋入了10億張公開可用的Instagram圖像,這些圖像未經過手動標記。但是,即使沒有通常在AI算法訓練中使用的標簽和注釋,SEER仍能夠自主地訓練數據集,不斷進行學習,并最終在諸如對象檢測之類的任務上達到最高的準確性。
圖片來自Facebook AI
這種被稱為自監督學習(self-supervised learning)的方法在AI領域已經很成熟:它由可以直接從給定信息中學習的系統組成,而不必依賴經過仔細標記的數據集來教他們如何執行諸如識別照片中的對象或翻譯文本之類的任務。
自監督式學習近來引起了很多關注,因為這意味著需要手工標記數據的工作要少很得,這對大多數研究人員而言是費時費力的工作。無需管理數據集的同時,自監督模型可以處理更大,更多樣化的數據集。
在某些領域,特別是自然語言處理中,該方法已經取得了突破。在數量越來越多的未標記文本上訓練算法已使諸如問答、機器翻譯、自然語言推理等應用程序取得了進展。
相反,計算機視覺尚未完全進入自監督的學習革命。正如Facebook AI Research的軟件工程師Priya Gopal解釋的那樣,SEER是該領域的首創。她告訴ZDNet:“與現有的在ImageNet數據集上訓練的計算機視覺的自監督模型相比,SEER是第一個可以隨機訓練互聯網上圖像上的完全自監督的計算機視覺模型?!?/p>
ImageNet是一個大規模數據庫,包含研究人員標記的數百萬張圖片,并向較大的計算機視覺社區開放,以促進AI的發展。
該項目的數據庫被Facebook的研究人員用作評估SEER性能的基準,他們發現自監督模型在諸如低空拍攝、物體檢測、分割和圖像分類等任務上優于最新的監督AI系統。
Goyal說:“通過僅對隨機圖像進行訓練,SEER優于現有的自監督模型。這一結果表明,我們不需要像ImageNet這樣的高度精選的數據集,對隨機圖像的自監督學習就可以產生非常高質量的模型?!?/p>
隨著自監督學習復雜度的提高,研究人員的工作并非沒有挑戰。在文本方面,AI模型的任務是為單詞賦予含義。但是對于圖像,該算法必須決定每個像素如何與一個概念相對應,同時考慮到其在不同圖片中拍攝角度、視圖和形狀的差別。
換句話說,研究人員需要大量數據,并且需要一個能夠從這種復雜的信息池中推導出所有可能的視覺概念。
為了完成此任務,Goyal和她的團隊從Facebook AI在自監督學習中的現有工作中改編了一種新算法,稱為SwAV,該算法將顯示相似的圖像聚集到單獨的組中??茖W家還設計了卷積網絡,也就是一種深度學習算法,算法對人腦中神經元的連接模式進行建模,根據重要性分配給圖像中的不同對象。
至少可以說,借助Instagram的10億張圖片的數據集,該系統的規模很大。Facebook的團隊使用了具有32GB RAM的V100 Nvidia GPU,并且隨著模型尺寸的增加,必須將模型放入可用的RAM中。但是Goyal解釋說,進一步的研究將對確保計算功能適應新系統很有用。
“隨著我們在越來越多的GPU上訓練模型,這些GPU之間的通信需要快速進行。可以通過開發軟件來解決給定的內存和運行時間帶來的挑戰?!彼f。
盡管仍有許多工作要做,但是,在可以將SEER應用于現實世界的用例之前,Goyal認為不應低估該技術的影響。她說:“借助SEER,我們現在可以通過大量隨機的互聯網上的圖像訓練大型模型,在計算機視覺方面取得更大的進步。”
“這一突破可以實現計算機視覺的自監督學習革命,類似于我們在自然語言處理文本中所看到的。”
在Facebook內部,SEER可用于各種計算機視覺任務,從自動生成圖像描述到幫助識別違反政策的內容。在公司外部,該技術還可用于圖像和元數據有限的領域,例如醫學成像。
Facebook的團隊呼吁開展更多工作,以將SEER推進到其下一階段的開發。作為研究的一部分,該團隊開發了一種基于PyTorch的全能庫,用于自監督學習,稱為VISSL,該庫是開源的,鼓勵更多的AI社區成員對該技術進行測試。
當被問及是否已通知Instagram用戶使用其圖像來訓練SEER或用戶是否有機會選擇其圖像不被用來學習時,Goyal指出,Facebook在其數據政策中告知Instagram帳戶持有人,其使用圖片等信息來支持研究,其中包括支持SEER的研究。也就是說,Facebook并不打算分享這些圖片或SEER模型本身,部分原因是該模型可能包含無意識的偏見。
Facebook在一篇博客文章中寫道:“自我監督學習一直是Facebook人工智能的重點,因為它使機器可以直接從世界上大量可用的信息中學習,而不僅僅是從專門為人工智能研究創建的訓練數據中學習。就像在其他研究領域一樣,自我監督學習對未來的計算機視覺有著不可思議的影響。消除了對人工注釋和元數據的需要,使計算機視覺社區能夠處理更大、更多樣化的數據集,從隨機公共圖像中學習,并可能減輕數據管理中出現的一些偏見。自我監督學習還可以幫助在圖像或元數據有限的領域(如醫學成像)對模型進行專門研究。無需人工提前標記,模型可以更快地創建和部署,能夠對快速變化的情況做出更快、更準確的反應?!?br /> 編輯:lyn
-
AI
+關注
關注
87文章
31711瀏覽量
270510
發布評論請先 登錄
相關推薦
評論