最美情侣中文字幕电影,在线麻豆精品传媒,在线网站高清黄,久久黄色视频

歡迎光臨散文網(wǎng) 會(huì)員登陸 & 注冊

比目前最先進(jìn)的模型輕30%!高效多機(jī)器人SLAM蒸餾描述符!

2023-07-11 13:16 作者:3D視覺工坊  | 我要投稿


作者:K.Fire | 來源:3D視覺工坊

在公眾號(hào)「3D視覺工坊」后臺(tái),回復(fù)「原論」即可獲取論文pdf。

添加微信:CV3d007,備注:SLAM,拉你入群。文末附行業(yè)細(xì)分群。

本文通過生成具有最小推理時(shí)間的緊湊且具有判別性的特征描述符來解決多機(jī)器人探索過程中保持低水平通信帶寬的同時(shí)進(jìn)行精確定位的問題;文中將描述符生成轉(zhuǎn)化為老師-學(xué)生框架下的學(xué)習(xí)問題。首先設(shè)計(jì)一個(gè)緊湊的學(xué)生網(wǎng)絡(luò),通過從預(yù)訓(xùn)練的大型教師模型中轉(zhuǎn)移知識(shí)來學(xué)習(xí)它。為了減少從教師到學(xué)生的描述符維度,文中提出了一種新的損失函數(shù),使知識(shí)在兩個(gè)不同維度的描述符之間轉(zhuǎn)移。

多機(jī)器人SLAM系統(tǒng)(MR-SLAM系統(tǒng))相比單機(jī)器人系統(tǒng)最主要的限制是:通信帶寬的限制。在MR-SLAM系統(tǒng)中,為了整合整個(gè)團(tuán)隊(duì)的所有軌跡,團(tuán)隊(duì)中的每個(gè)機(jī)器人都需要共享其關(guān)鍵幀數(shù)據(jù)(包括關(guān)鍵幀姿態(tài)和觀察到的特征點(diǎn)),以處理機(jī)器人間的閉環(huán)和全局定位。這種類型的數(shù)據(jù)交換占用了很高的通信容量,很可能降低實(shí)時(shí)性能。

一些工程上的解決方案比如降低頻率、減少關(guān)鍵點(diǎn)數(shù)量可以降低通信帶寬,但會(huì)導(dǎo)致定位精度降低;提高精度則會(huì)導(dǎo)致帶寬升高;如果降低描述符維度會(huì)導(dǎo)致匹配性能變差。

卷積神經(jīng)網(wǎng)絡(luò)(CNN)方法已經(jīng)顯示出對手工描述符的優(yōu)越性能,但基于CNN的方法多適用于單機(jī)器人系統(tǒng)(SR-SLAM),高模型復(fù)雜性和描述符維度是阻礙基于學(xué)習(xí)的方法在MR-SLAM中應(yīng)用的兩個(gè)主要障礙。

本文提出了一個(gè)基于知識(shí)蒸餾(KD)的描述符學(xué)習(xí)框架。該框架是一個(gè)經(jīng)典的師生組合,其中學(xué)生是我們要學(xué)習(xí)的緊湊模型,而教師是一個(gè)預(yù)訓(xùn)練的更大的網(wǎng)絡(luò),其預(yù)測用于監(jiān)督學(xué)生的學(xué)習(xí)。為了解決老師和學(xué)生之間輸出維度的不同,本文還設(shè)計(jì)了一種新的損失函數(shù)。

本文模型與傳統(tǒng)模型的比較如下圖所示:

本文主要貢獻(xiàn)如下:

  • 設(shè)計(jì)了一個(gè)師生模型來生成緊湊的二進(jìn)制描述符。

  • 提出了一種新的基于距離的蒸餾損失,它允許具有不同輸出維度的模型之間的知識(shí)轉(zhuǎn)移。

  • 開發(fā)了一個(gè)基于新描述符模型的MR-SLAM系統(tǒng)作為評(píng)估平臺(tái)。

本文類似于主流的CNN方法,通過Siamese框架結(jié)構(gòu)學(xué)習(xí)描述符。它是一個(gè)雙胞胎模型,在訓(xùn)練過程中共享相同的權(quán)重。在Siamese框架下,每個(gè)訓(xùn)練階段都準(zhǔn)備兩批。批次中的每個(gè)圖像塊對另一個(gè)都是負(fù)的,同一時(shí)代的兩個(gè)批是一一對應(yīng)的。

它有兩個(gè)主要的損耗用于訓(xùn)練:

  1. Triplet loss:它強(qiáng)制匹配的對盡可能接近,而不匹配的對盡可能遠(yuǎn)。最近的方法大多遵循HardNet策略,只提取包含最小距離的負(fù)對進(jìn)行訓(xùn)練:

為當(dāng)前實(shí)值描述子,為對應(yīng)的正描述子,為最接近的負(fù)描述子,n為批大小。dis計(jì)算兩個(gè)描述符之間的L2距離。

  1. Binarization loss:它旨在最小化實(shí)值描述符與對應(yīng)的二進(jìn)制描述符之間的差異:

B(k)為二進(jìn)制描述子B的第k個(gè)值,R(k)為實(shí)值描述子R的第k個(gè)值,D為描述子的維數(shù)。

如上圖所示,所有的老師卷積層之后是濾波響應(yīng)歸一化(FRN)和閾值線性單元(TLU)。批處理歸一化(BN)和l2歸一化放在最后的卷積層之后。

學(xué)生卷積層第一部分是深度卷積,它對每個(gè)輸入通道應(yīng)用一個(gè)卷積濾波器。第二部分是1×1卷積,稱為逐點(diǎn)卷積。

基于該體系結(jié)構(gòu),每個(gè)模型都能夠?qū)D像補(bǔ)丁嵌入到實(shí)值描述符中。為了得到二進(jìn)制描述符,使用以下方程對實(shí)值描述符進(jìn)行二值化:

它由一個(gè)實(shí)值項(xiàng)和一個(gè)二元值項(xiàng)組成,它們分別強(qiáng)制學(xué)生模型的描述符批在實(shí)值和二元空間上具有與教師模型相似的分布。

實(shí)值項(xiàng)負(fù)描述符損失項(xiàng):

其中是相對于當(dāng)前描述符的負(fù)描述符的總數(shù)。和分別是教師和學(xué)生產(chǎn)生的當(dāng)前重估描述符,和是相應(yīng)的負(fù)描述符。此外,由于教師和學(xué)生之間的維度不同,我們增加了一個(gè)系數(shù)λr,設(shè)置為0.95來調(diào)整教師側(cè)距離的尺度

二值項(xiàng)負(fù)描述符損失項(xiàng):

其中λb是縮放系數(shù),通常設(shè)置為兩個(gè)模型的輸出維度之間的比率:Ds/Dt。在訓(xùn)練階段的到二進(jìn)制描述符計(jì)算公式:

這里是防止被0整除的系數(shù),設(shè)為1e-5。

最后,將這兩個(gè)損失項(xiàng)合并為蒸餾損失函數(shù):

其中是二值項(xiàng)的加權(quán)系數(shù)。

第一階段,使用學(xué)習(xí)率為0.01的ADAM優(yōu)化器訓(xùn)練教師模型。這個(gè)過程使模型能夠生成高質(zhì)量的128維描述符。基本損耗函數(shù)為:

其中是控制二值化損失的加權(quán)系數(shù)。

在第二階段,在老師的監(jiān)督下訓(xùn)練學(xué)生模型,學(xué)習(xí)率設(shè)為0.01。因此,目標(biāo)函數(shù)既包括基本損失Lbasic,也包括蒸餾損失Ldistillation,可表示為:

其中為蒸餾損失加權(quán)系數(shù),在訓(xùn)練時(shí)設(shè)置為2。

使用UBC數(shù)據(jù)集:整個(gè)數(shù)據(jù)集包含大約400k個(gè)64×64個(gè)帶標(biāo)簽的補(bǔ)丁,在一個(gè)子集上訓(xùn)練模型,并在另外兩個(gè)子集上測試它。通過報(bào)告95%召回率(FPR95)的假陽性率來展示匹配性能。

采用幾種方法作為基準(zhǔn)方法。第一類是基于cnn的方法,包括HardNet、SOSNet、HyNet等。第二類方法是一些傳統(tǒng)的方法,包括SIFT、BRIEF、LDAHash、BinBoost。還報(bào)告了每種CNN方法的參數(shù)數(shù)量和時(shí)間消耗(生成500個(gè)描述符),以可視化描述符生成效率。

此外,為了測試泛化能力,文章評(píng)估了在HPatches數(shù)據(jù)集上訓(xùn)練并在UBC數(shù)據(jù)集上測試的CNN方法的性能。

實(shí)驗(yàn)結(jié)果如下表所示:

  1. 由表1可以看出,基于cnn的方法明顯優(yōu)于傳統(tǒng)方法。另一方面,匹配結(jié)果表明,即使沒有蒸餾,所提出的方法也可以與大多數(shù)基于cnn的方法相媲美。此外,在KD方法的幫助下,文中的網(wǎng)絡(luò)能夠優(yōu)于所有基線方法。

  2. 由表2可以看出,文中的模型比基線CNN模型輕30%左右,運(yùn)行速度也比基線方法快。

  3. 由表3可以看出,因?yàn)楸M管在不同的數(shù)據(jù)集上訓(xùn)練,但它的性能與CNN方法相當(dāng)。

MR-VINS的詳細(xì)框架如下圖所示:

整個(gè)系統(tǒng)采用集中式架構(gòu),即MR-SLAM系統(tǒng)由多個(gè)機(jī)器人和一個(gè)中央服務(wù)器組成。將視覺慣性里程計(jì)模塊放在機(jī)器人上,而將后端模塊放在服務(wù)器上。在此框架的基礎(chǔ)上,在機(jī)器人一側(cè)添加了CNN描述符模型,主要將描述符用于循環(huán)閉合和全局定位。作者將描述符模型放在機(jī)器人通信模塊之前,以便描述符可以集成到發(fā)送到服務(wù)器的關(guān)鍵幀消息中。

使用公共EuRoC數(shù)據(jù)集,采用定位精度和通信帶寬兩項(xiàng)指標(biāo)進(jìn)行評(píng)價(jià)。

SLAM精度的定量結(jié)果如下表所示。

與原始MR-VINS相比,可以看出,基于cnn的方法能夠提高定位性能,尤其是MR-SLAM。此外,在單機(jī)器人和多機(jī)器人任務(wù)上,文中的方法總體上達(dá)到了與CNN方法相當(dāng)?shù)木取H欢?,由于SLAM的性能不僅僅取決于圖像匹配,定位精度優(yōu)勢并不像特征匹配結(jié)果那么明顯。

下表顯示了不同SLAM系統(tǒng)的帶寬。

結(jié)果表明,文中的方法比原來的MR-VINS和CCM-SLAM具有更低的帶寬。由于文中模型的消息頻率是CCM-SLAM的3倍,所以文中模型的關(guān)鍵幀信息的大小比CCM-SLAM小10倍左右。窄帶寬歸因于文中的描述符極其緊湊的尺寸(普通BRIEF描述符的1/4)。此外,由于關(guān)鍵幀尺寸緊湊,就不需要在MR-SLAM任務(wù)期間專門降低關(guān)鍵幀頻率。

在本文中,探討了如何學(xué)習(xí)MR-SLAM的緊湊描述符的問題。本文提出了一個(gè)師生框架,該框架利用緊湊的學(xué)生模型來估計(jì)低維描述符。由于教師和學(xué)生之間的輸出維度不同,作者提出了一個(gè)基于距離的蒸餾損失函數(shù),使知識(shí)在不同的維度描述符之間蒸餾傳遞。通過特征匹配實(shí)驗(yàn)和MR-SLAM系統(tǒng)的實(shí)驗(yàn),驗(yàn)證了本文所提出算法的有效性。

加群

目前工坊已經(jīng)建立了3D視覺方向多個(gè)社群,包括SLAM、工業(yè)3D視覺、自動(dòng)駕駛方向,細(xì)分群包括:[工業(yè)方向]三維點(diǎn)云、結(jié)構(gòu)光、機(jī)械臂、缺陷檢測、三維測量、TOF、相機(jī)標(biāo)定、綜合群;[SLAM方向]多傳感器融合、ORB-SLAM、激光SLAM、機(jī)器人導(dǎo)航、RTK|GPS|UWB等傳感器交流群、SLAM綜合討論群;[自動(dòng)駕駛方向]深度估計(jì)、Transformer、毫米波|激光雷達(dá)|視覺攝像頭傳感器討論群、多傳感器標(biāo)定、自動(dòng)駕駛綜合群等。[三維重建方向]NeRF、colmap、OpenMVS等。除了這些,還有求職、硬件選型、視覺產(chǎn)品落地等交流群。大家可以添加小助理微信: CV3d007,備注:加群+方向+學(xué)校|公司, 小助理會(huì)拉你入群。

比目前最先進(jìn)的模型輕30%!高效多機(jī)器人SLAM蒸餾描述符!的評(píng)論 (共 條)

分享到微博請遵守國家法律
永春县| 江西省| 台东县| 凤冈县| 华蓥市| 牙克石市| 临湘市| 油尖旺区| 旌德县| 筠连县| 昆明市| 榆林市| 郴州市| 淮安市| 英山县| 南投市| 台山市| 宾阳县| 梓潼县| 鄂州市| 新邵县| 泽州县| 徐州市| 鸡西市| 乌拉特后旗| 乌鲁木齐市| 化德县| 定远县| 会昌县| 富蕴县| 承德县| 常州市| 诸城市| 邹城市| 哈巴河县| 石泉县| 瓦房店市| 桃园市| 红河县| 萨嘎县| 高碑店市|