Facebook Inc.開放了一個(gè)名為AI Habitat的平臺(tái)的源代碼

諸如工業(yè)機(jī)器人之類的機(jī)器內(nèi)部的人工智能模型要求能夠與環(huán)境安全有效地交互。但是在現(xiàn)實(shí)環(huán)境中甚至是無風(fēng)險(xiǎn)的訓(xùn)練空間中訓(xùn)練AI往往是不切實(shí)際的，因?yàn)榻虒?dǎo)復(fù)雜的神經(jīng)網(wǎng)絡(luò)需要大量的時(shí)間。

為了幫助加快這一進(jìn)程，F(xiàn)acebook Inc.今天開放了一個(gè)名為AI Habitat的平臺(tái)的源代碼。它是一個(gè)模擬器，使研究人員可以在虛擬環(huán)境中一次執(zhí)行許多培訓(xùn)課程。

AI Habitat可能不是第一個(gè)考慮到機(jī)器學(xué)習(xí)項(xiàng)目構(gòu)建的模擬器，但是它肯定是最快的模擬器之一。據(jù)Facebook稱，內(nèi)部測試表明該平臺(tái)僅使用一張圖形卡就可以以10，000幀/秒的速度運(yùn)行虛擬培訓(xùn)環(huán)境。社交網(wǎng)絡(luò)嘗試的其他仿真引擎每秒只能達(dá)到50到100幀。

人居AI的四位工程師Manolis Savva，Abhishek Kadian，Oleksandr Maksymets和Dhruv Batra在博客中寫道 ，他們的團(tuán)隊(duì)“將訓(xùn)練速度推上了最先進(jìn)的狀態(tài)，使模擬器能夠渲染出比以前的模擬器快得多?！?Facebook聲稱使用AI Habitat可以在數(shù)小時(shí)內(nèi)完成使用其他工具進(jìn)行數(shù)月的實(shí)驗(yàn)。

該公司構(gòu)建了可在廣泛的AI項(xiàng)目中使用的平臺(tái)。Facebook將其模塊化，以便研究人員可以根據(jù)需要交換關(guān)鍵組件，并為用于訓(xùn)練AI模型的幾個(gè)通用開源數(shù)據(jù)集實(shí)現(xiàn)了本機(jī)支持。

該社交網(wǎng)絡(luò)開源了自己的培訓(xùn)數(shù)據(jù)集，該數(shù)據(jù)集專門設(shè)計(jì)用于與AI Habitat一起使用。該存儲(chǔ)庫被稱為副本，包括18個(gè)虛擬室內(nèi)環(huán)境，從辦公室會(huì)議室到兩層樓的房屋。

副本之所以被命名，是因?yàn)樗盀檎鎸?shí)空間的3D重構(gòu)的真實(shí)性和質(zhì)量設(shè)定了新標(biāo)準(zhǔn)，”致力于將虛擬環(huán)境整合在一起的Facebook研究人員之一朱利安·斯特勞布（Julian Straub）解釋說。它們不是在圖形設(shè)計(jì)工具中創(chuàng)建的效果圖，而是對(duì)社交網(wǎng)絡(luò)使用自定義攝像頭技術(shù)拍攝的真實(shí)空間的掃描。

該公司利用紅外深度捕獲系統(tǒng)來記錄每種環(huán)境中物體的確切形狀。然后以高分辨率重建書籍，椅子和窗戶等物品，并保留其紋理，以確保AI模型暴露于逼真的訓(xùn)練環(huán)境中。Facebook的研究人員還添加了描述每個(gè)對(duì)象的標(biāo)簽。

研究人員可以使用AI Habitat訓(xùn)練神經(jīng)網(wǎng)絡(luò)來執(zhí)行相當(dāng)復(fù)雜的多步驟任務(wù)，這些任務(wù)不僅需要能夠看到而且還要了解其周圍環(huán)境。例如，家用機(jī)器人可以配備處理“檢查我的筆記本電腦是否在廚房的桌子上”之類的請(qǐng)求的功能。但是AI Habitat中的培訓(xùn)環(huán)境不是交互式的，這意味著一旦發(fā)現(xiàn)筆記本電腦，就不可能教機(jī)器人如何獲取筆記本電腦。