跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

LookWhen? 透過學習何時、何地以及計算什麼來實現快速影片識別

文章摘要

LookWhen 是一種新的影片識別框架,透過選擇-提取機制學習何時、何地以及計算什麼,從而在不處理所有令牌的情況下近似完整影片表示,顯著提升效率。實驗表明,LookWhen 在多個資料集上實現了更好的準確率-計算平衡,比 InternVideo2-B 快6.7倍。

來源arXiv Computer Vision作者: Ali Salamatian, Anthony Fuller, Pritam Sarkar, James R. Green, Leonid Sigal, Evan Shelhamer
LookWhen? 透過學習何時、何地以及計算什麼來實現快速影片識別
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

影片識別領域長期以來由 Transformer 模型主導,它們將影片分割成令牌,處理成本呈超線性增長。然而,影片資料充滿冗餘,因此有必要質疑這種高昂計算開銷的必要性。為此,研究人員提出了 LookWhen,一種選擇器-提取器框架,將影片識別任務分解為學習何時、何地以及計算什麼。

LookWhen 的設計包含兩個關鍵元件:一個淺層選擇器和一個深層提取器。選擇器接收縮小後的影片,快速對所有時空令牌進行評分;提取器則僅處理評分最高的前 K 個令牌,從而近似完整影片表示,而無需真正處理所有令牌。這種方法大幅降低了計算量,因為只有一小部分令牌被送入深層網路。

訓練 LookWhen 面臨一個關鍵挑戰:如何為選擇和提取提供有效的監督訊號。針對選擇預訓練,團隊引入了一種基於表示唯一性分數的排序方法,透過簡單的最近鄰距離來衡量每個令牌的獨特性。對於提取預訓練,則採用知識蒸餾策略,分別從影片教師模型和影像教師模型中學習;影像教師模型透過歸一化幀表示來捕捉影片內部的時序變化。這些策略使得選擇器-提取器能夠學習到通用且高效的表示,適用於特徵提取或任務微調。

研究團隊在 Kinetics-400、SSv2、Epic-Kitchens、Diving48、Jester 和 Charades 六個資料集上進行了全面的實驗。結果令人印象深刻:LookWhen 在準確率與計算量(FLOPs)之間取得了更好的平衡,在6個任務×2種設定的12種情況下,有9種實現了 Pareto 優勢,其餘3種也基本持平。在實際吞吐量測試中,LookWhen 的效率更為突出,在同等準確率下比 InternVideo2-B 快6.7倍。這意味著在實際應用中,LookWhen 可以在更短的時間內處理大量影片,同時保持高識別精度。

這一成果為高效影片識別提供了新的方向,尤其適用於資源受限的場景,如移動裝置或即時監控系統。未來,LookWhen 的設計思路可以擴充套件到其他需要高效處理時空訊號的任務,為計算機視覺領域帶來更廣泛的節能與加速效益。

展開要點與分析

文章情報

工程師進階

要點

  • LookWhen 將影片識別分解為學習何時、何地以及計算什麼。
  • 使用淺層選擇器快速評分所有時空令牌,深層提取器僅處理前 K 個令牌。
  • 預訓練策略包括基於唯一性分數的選擇蒸餾和影片/影像教師提取蒸餾。
  • 在6個資料集上,LookWhen 在準確率-FLOPs 上 Pareto 優於大多數基線,吞吐量提升顯著。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。