影片識別領域長期以來由 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 的設計思路可以擴充套件到其他需要高效處理時空訊號的任務,為計算機視覺領域帶來更廣泛的節能與加速效益。