47.4mAP!最強Anchor-free目標檢測網路:SAPD
- 2019 年 12 月 10 日
- 筆記
前言
最近目標檢測相關的論文非常多,可見Object Detection還是有很多工作(坑)可以做(填)的。上次 Amusi 就整理了近期值得關注的目標檢測論文,詳見:一文看盡8篇目標檢測最新論文(EfficientDet/EdgeNet/ASFF/RoIMix/SCL/EFGRNet等)
縱覽2019年,目標檢測除了NAS分支外,Anchor-free算是佔據"半壁江山"。其中比較亮眼的工作有很多,如:FSAF、FCOS、CenterNet、FoveaBox和FreeAnchor等
關於Anchor-free更多的解讀,詳見下面兩篇文章:
目標檢測:Anchor-Free時代
重磅!13篇基於Anchor free的目標檢測方法
SAPD:Soft Anchor-Point Detector

時間:2019年11月28日
作者團隊:卡耐基梅隆大學(CMU)
論文鏈接:https://arxiv.org/abs/1911.12448
近來,anchor-free 目標檢測器在準確性和速度方面都顯著的超越 anchor-based 的檢測器。在SAPD這個工作中,作者旨在為anchor-free 檢測器尋找速度(FPS)和準確性(mAP)之間的新平衡。

Single-model single-scale speed (ms) vs. accuracy (AP) on COCO test-dev
SAPD 研究了兩個問題:
1)如何使 anchor-free 目標檢測的"head"更好?
2)如何更好地利用特徵金字塔的特性?
作者將注意力偏差(attention bias)和特徵選擇分別確定為上述兩個問題的主要痛點。於是作者提出使用一種新的訓練策略來解決這些問題,該策略具有兩種 soften 優化方法,即:
SW:soft-weighted anchor points
SS:soft-selected pyramid levels
為了評估演算法的有效性,作者訓練了一個稱為SAPD:Soft Anchor-Point Detector的single-stage anchor-free 檢測器。實驗表明,優雅的 SAPD 將速度/準確度的 trade-off 達到了一個新的水平,超過了之前的 anchor-based,anchor-free,single-stage和 multi-stage 檢測器(這裡其實都是各家吹捧的點,各種打敗SOTA,大家聽聽就好,因為實際上目前並沒有出現mAP和FPS都是SOTA的檢測演算法)。
SAPD 可以在 COCO 上實現 47.4% mAP(單模型單尺度)。作者稱:SAPD最快的版本可以比其他具有相同的精度的檢測器速度快5倍!
網路結構

anchor-point detector

訓練策略:soft-weighted anchor points 和 soft-selected pyramid levels

實驗結果



為了方便下載,我已經將 SAPD 論文打包,在 CVer公眾號 後台回復:20191206 即可獲得打包鏈接。同時論文會同步推送到 github上,歡迎大家 star/fork(點擊閱讀原文,也可直接訪問):
https://github.com/amusi/daily-paper-computer-vision
如果各位 CVer 喜歡這樣的最強系列,請給這篇文章點個 在看