
銀河水滴科技有限公司(以下簡(jiǎn)稱為“銀河水滴”)源自于中科院自動(dòng)化所,是步態(tài)識(shí)別技術(shù)和產(chǎn)業(yè)的引領(lǐng)者,致力于將此技術(shù)與行業(yè)應(yīng)用深度融合。在對(duì)步態(tài)識(shí)別AI模型進(jìn)行訓(xùn)練時(shí),銀河水滴部署了包含8塊NVIDIAV100數(shù)據(jù)中心GPU的NVIDIADGX-1系統(tǒng),通過(guò)強(qiáng)勁算力處理多階段、多種類、多用途的大量數(shù)據(jù),加速步態(tài)識(shí)別技術(shù)在刑偵、安防、人群密度監(jiān)控等領(lǐng)域的應(yīng)用落地。
復(fù)雜環(huán)境亟需高效方案
步態(tài)識(shí)別面臨著極具挑戰(zhàn)的復(fù)雜環(huán)境——遠(yuǎn)距離、跨視角、跨著裝、低照度、全天候,是極具挑戰(zhàn)、但又及其重要的科技難題,是公認(rèn)的當(dāng)前視頻大數(shù)據(jù)解析與應(yīng)用的核心技術(shù)之一。由于其數(shù)據(jù)環(huán)境復(fù)雜,因而所需要處理的數(shù)據(jù)種類繁多。
為了在步態(tài)識(shí)別過(guò)程中通過(guò)深度學(xué)習(xí)方法處理視角、著裝、攜帶物等多種行人特征的同時(shí),保持所生成步態(tài)序列的不變性,銀河水滴所需處理的數(shù)據(jù)量十分龐大,因而AI模型的訓(xùn)練數(shù)據(jù)吞吐量大、整體訓(xùn)練速度偏慢,亟需尋找更強(qiáng)大的GPU解決方案以更高效地從步態(tài)序列中提取更具魯棒性的特征。
40%時(shí)間成本的差距
在步態(tài)識(shí)別AI模型訓(xùn)練中,需要通過(guò)GPU對(duì)大量數(shù)據(jù)進(jìn)行傳輸與計(jì)算。英邁中國(guó)及其資深合作伙伴寶德協(xié)助銀河水滴試用了NVIDIADGX-1系統(tǒng),NVIDIADGX-1所具有的高帶寬可以大幅加速數(shù)據(jù)搬運(yùn)的過(guò)程給銀河水滴留下了深刻的印象。最終,為了進(jìn)一步加速步態(tài)識(shí)別AI模型訓(xùn)練,銀河水滴選擇部署了NVIDIADGX-1深度學(xué)習(xí)服務(wù)器以滿足高計(jì)算需求。
NVIDIADGX-1是款深度學(xué)習(xí)系統(tǒng),專為實(shí)現(xiàn)吞吐量和互聯(lián)帶寬構(gòu)建,可極提升神經(jīng)絡(luò)訓(xùn)練性能。其系統(tǒng)核組件是8個(gè)NVIDIAV100GPU,通過(guò)混合體NVlink絡(luò)拓?fù)浠ヂ?lián)。除了這8個(gè)GPU之外,NVIDIADGX-1還內(nèi)置兩塊CPU以供啟動(dòng)、存儲(chǔ)管理及深度學(xué)習(xí)框架協(xié)調(diào)之。NVIDIADGX-1內(nèi)嵌于3機(jī)架單位(3U)機(jī)箱中,集電源、冷卻、絡(luò)、多系統(tǒng)互聯(lián)及SSD件系統(tǒng)緩存于體,各組件相互協(xié)調(diào),能夠優(yōu)化吞吐量和深度學(xué)習(xí)訓(xùn)練時(shí)間。通過(guò)部署NVIDIADGX-1系統(tǒng),銀河水滴取得了比單獨(dú)部署GPU更強(qiáng)大的性能。
在模型訓(xùn)練過(guò)程中銀河水滴的技術(shù)人員發(fā)現(xiàn),NVIDIADGX-1可以協(xié)助充分發(fā)揮NVIDIA數(shù)據(jù)中心GPU的潛力。銀河水滴在對(duì)其模型進(jìn)行每100次迭代時(shí),此前通過(guò)單獨(dú)采用配置了8塊NVIDIAV100GPU的服務(wù)器所需的訓(xùn)練時(shí)間是2分40秒,而在包含8塊NVIDIAV100數(shù)據(jù)中心GPU的NVIDIADGX-1系統(tǒng)中,這一過(guò)程則只需花費(fèi)1分40秒,縮短了近40%的訓(xùn)練時(shí)間。
NVIDIA助力AI行業(yè)加速落地
目前,銀河水滴的步態(tài)識(shí)別技術(shù)已在全國(guó)各地幾十個(gè)城市落地,在對(duì)步態(tài)識(shí)別AI模型進(jìn)行訓(xùn)練時(shí),部署了包含8塊NVIDIAV100數(shù)據(jù)中心GPU的NVIDIADGX-1系統(tǒng),通過(guò)強(qiáng)勁算力處理多階段、多種類、多用途的大量數(shù)據(jù),加速了銀河水滴步態(tài)識(shí)別技術(shù)的應(yīng)用落地。DGX-1系統(tǒng)所具有的革命性AI性能也將繼續(xù)提高銀河水滴的AI模型訓(xùn)練效率,以助力經(jīng)過(guò)充分迭代的深度學(xué)習(xí)模型應(yīng)用至更多場(chǎng)景中。
今后,銀河水滴將持續(xù)深耕智能視頻分析領(lǐng)域,以先進(jìn)的人工智能技術(shù)提升視頻數(shù)據(jù)價(jià)值,領(lǐng)跑數(shù)智視頻新基建大賽道。而英邁將繼續(xù)致力于向合作伙伴及客戶提供更優(yōu)的深度學(xué)習(xí)解決方案,結(jié)合NVIDIA數(shù)據(jù)中心GPU的不斷進(jìn)化及DGX系統(tǒng)的升級(jí),通過(guò)AI基礎(chǔ)架構(gòu)的性能飛躍助力各行各業(yè),幫助客戶部署即插即的超高性能深度神經(jīng)絡(luò)訓(xùn)練平臺(tái),引領(lǐng)客戶在深度學(xué)習(xí)工作負(fù)載方面保持競(jìng)爭(zhēng)優(yōu)勢(shì)。