近日,阿里雲正式發佈面向自動駕駛領域模型的訓練、推理加速框架——PAI-TurboX,旨在提升感知、規劃控制乃至世界模型的訓推效率。該框架在多個行業模型的訓練任務中展現出了顯著優勢,可縮短高達50%的時間。

PAI-TurboX框架的應用範圍廣泛,涵蓋了多模態數據預處理、離線大規模模型訓練以及實時智駕推理等多個環節,爲自動駕駛、具身智能等領域提供了全面解決方案。目前,該框架已成功應用於多家車企,助力其提升自動駕駛技術的研發和應用效率。

在系統側,PAI-TurboX通過優化CPU親和性、動態編譯、流水線並行等策略,顯著提升了模型的訓練推理效率。而在數據側,該框架提出了高性能的DataLoader引擎,並優化了數據預處理流程,實現了智能訓練樣本分組,有效提升了數據處理效率。

微信截圖_20250623160139.png

此外,PAI-TurboX還提供了算子優化和量化等能力,可進一步減少訓練階段的訪存延遲,提升吞吐效率。在推理任務中,該框架能在保障精度的同時降低計算開銷與內存帶寬需求,實現異構平臺下的高性能推理部署。

實測結果顯示,在自動駕駛的3D物體檢測模型BEVFusion訓練任務中,PAI-TurboX可將訓練時間縮短58.5%;在實時在線矢量化高精地圖構建模型MapTR訓練任務中,可將訓練時間縮短53%;在端到端自動駕駛模型SparseDrive訓練任務中,同樣在感知模塊訓練和聯合訓練兩個階段獲得了明顯的速度提升。

阿里雲人工智能平臺PAI自2016年誕生以來,已累計服務超過10萬家企業客戶及數百萬AI開發者,支撐了阿里雲百鍊、魔搭社區等MaaS服務及社區的發展。此次PAI-TurboX框架的推出,無疑將進一步推動自動駕駛領域的技術創新和應用落地。