視覺感知是智能掃地機(jī)器人實(shí)現(xiàn)環(huán)境識(shí)別、自主避障、地毯檢測(cè)、防跌落等核心功能的核心支撐,而深度學(xué)習(xí)神經(jīng)網(wǎng)絡(luò)則是視覺感知算法的核心載體。但掃地機(jī)器人搭載的嵌入式平臺(tái),普遍存在算力有限、功耗嚴(yán)格、存儲(chǔ)容量小、散熱條件有限的客觀約束,無(wú)法直接運(yùn)行計(jì)算機(jī)視覺領(lǐng)域常規(guī)的深度神經(jīng)網(wǎng)絡(luò)模型。在此背景下,面向嵌入式平臺(tái)的輕量化視覺神經(jīng)網(wǎng)絡(luò)應(yīng)運(yùn)而生,通過結(jié)構(gòu)精簡(jiǎn)、算法優(yōu)化、硬件協(xié)同等手段,在保證視覺識(shí)別精度的前提下,大幅降低模型算力占用、參數(shù)量與推理時(shí)延,讓高性能視覺感知能力落地于家用掃地機(jī)器人。本文從掃地機(jī)器人嵌入式平臺(tái)的特性出發(fā),拆解輕量化視覺神經(jīng)網(wǎng)絡(luò)的設(shè)計(jì)原則、核心優(yōu)化技術(shù)、典型網(wǎng)絡(luò)架構(gòu)、場(chǎng)景適配方法及落地挑戰(zhàn),全面解析其在掃地機(jī)器人領(lǐng)域的應(yīng)用邏輯與技術(shù)價(jià)值。
掃地機(jī)器人的嵌入式主控平臺(tái),與服務(wù)器、PC端、高端手機(jī)等設(shè)備存在顯著差異,這些硬件特性直接決定了輕量化視覺神經(jīng)網(wǎng)絡(luò)的設(shè)計(jì)方向,也是模型優(yōu)化遵循的前提條件。從算力層面來看,掃地機(jī)器人主控芯片多為中端ARM架構(gòu)或?qū)S肁IoT芯片,AI算力通常僅支持TOPS級(jí)別,遠(yuǎn)低于高端計(jì)算平臺(tái),無(wú)法承載大規(guī)模卷積運(yùn)算;從功耗層面來看,設(shè)備依靠電池供電,續(xù)航時(shí)長(zhǎng)是核心指標(biāo),神經(jīng)網(wǎng)絡(luò)推理功耗控制在極低范圍,避免過度耗電影響清潔作業(yè);從存儲(chǔ)層面來看,設(shè)備閃存容量有限,模型文件體積需嚴(yán)格壓縮,同時(shí)運(yùn)行內(nèi)存也無(wú)法支撐大模型的特征圖緩存;從實(shí)時(shí)性層面來看,掃地機(jī)器人移動(dòng)過程中需快速完成環(huán)境感知與決策,視覺推理時(shí)延需控制在毫秒級(jí)別,否則會(huì)出現(xiàn)避障滯后、漏識(shí)別等問題。
與此同時(shí),掃地機(jī)器人的視覺感知任務(wù)具有明確的場(chǎng)景針對(duì)性,無(wú)需處理復(fù)雜的通用視覺任務(wù),只需聚焦家居環(huán)境下的障礙物識(shí)別、地毯分類、落差檢測(cè)、語(yǔ)義分割等有限任務(wù),這為神經(jīng)網(wǎng)絡(luò)輕量化提供了應(yīng)用基礎(chǔ)。輕量化視覺神經(jīng)網(wǎng)絡(luò)的核心目標(biāo),是在滿足家居視覺識(shí)別精度的基礎(chǔ)上,實(shí)現(xiàn)算力、功耗、時(shí)延、存儲(chǔ)的多重適配,讓模型在嵌入式平臺(tái)上穩(wěn)定、高效運(yùn)行,支撐機(jī)器人完成全場(chǎng)景自主清潔。
針對(duì)掃地機(jī)器人嵌入式平臺(tái)的約束與視覺任務(wù)特性,輕量化視覺神經(jīng)網(wǎng)絡(luò)需遵循四大核心設(shè)計(jì)原則,平衡模型性能與硬件適配性,實(shí)現(xiàn)精準(zhǔn)感知與高效運(yùn)行的統(tǒng)一。
任務(wù)導(dǎo)向精簡(jiǎn),摒棄通用視覺模型的冗余結(jié)構(gòu),僅保留家居場(chǎng)景所需的特征提取能力。掃地機(jī)器人視覺任務(wù)聚焦于固定目標(biāo)識(shí)別與簡(jiǎn)單分割,無(wú)需復(fù)雜的深層網(wǎng)絡(luò)結(jié)構(gòu),通過縮減網(wǎng)絡(luò)層數(shù)、減少通道數(shù)、去除無(wú)關(guān)分支,聚焦障礙物輪廓、地面紋理、物體紋理等關(guān)鍵特征提取,降低無(wú)效運(yùn)算。
算力功耗均衡,優(yōu)先選用低算力占用的網(wǎng)絡(luò)算子與結(jié)構(gòu),控制模型推理功耗。避免使用大尺寸卷積核、全連接層等高運(yùn)算量模塊,采用輕量級(jí)算子替代常規(guī)運(yùn)算,同時(shí)優(yōu)化模型推理流程,減少中間特征圖緩存,降低內(nèi)存占用與功耗消耗。
精度損失可控,輕量化過程中嚴(yán)格把控識(shí)別精度,避免因過度壓縮導(dǎo)致漏檢、誤檢。通過精細(xì)化調(diào)參、特征增強(qiáng)、數(shù)據(jù)增強(qiáng)等方式,彌補(bǔ)模型精簡(jiǎn)帶來的精度下降,保證家居場(chǎng)景下各類目標(biāo)的識(shí)別準(zhǔn)確率,滿足清潔作業(yè)的實(shí)用需求。
第四,硬件協(xié)同適配,結(jié)合掃地機(jī)器人嵌入式芯片的架構(gòu)特性,優(yōu)化模型結(jié)構(gòu)與推理方式。針對(duì)芯片的AI加速單元、內(nèi)存帶寬、指令集特點(diǎn),設(shè)計(jì)匹配的網(wǎng)絡(luò)結(jié)構(gòu),提升硬件利用率,讓模型推理速度與平臺(tái)算力高度適配。