擴散模型不但在生成任務上非常成功，這次在目標檢測任務上，更是超越了成熟的目標檢測器。

擴散模型（ Diffusion Model ）作為深度生成模型中的新 SOTA，已然在圖像生成任務中超越了原 SOTA：例如 GAN，并且在諸多應用領域都有出色的表現(xiàn)，如計算機視覺，NLP、分子圖建模、時間序列建模等。

近日，來自香港大學的羅平團隊、騰訊 AI Lab 的研究者聯(lián)合提出一種新框架 DiffusionDet，將擴散模型應用于目標檢測。據(jù)了解，還沒有研究可以成功地將擴散模型應用于目標檢測，可以說這是第一個采用擴散模型進行目標檢測的工作。

DiffusionDet 的性能如何呢？在 MS-COCO 數(shù)據(jù)集上進行評估，使用 ResNet-50 作為骨干，在單一采樣 step 下，DiffusionDet 實現(xiàn) 45.5 AP，顯著優(yōu)于 Faster R-CNN (40.2 AP)， DETR (42.0 AP)，并與 Sparse R-CNN (45.0 AP)相當。通過增加采樣 step 的數(shù)量，進一步將 DiffusionDet 性能提高到 46.2 AP。此外，在 LVIS 數(shù)據(jù)集上，DiffusionDet 也表現(xiàn)良好，使用 swin - base 作為骨干實現(xiàn)了 42.1 AP。

DiffusionDet: Diffusion Model for Object Detection

論文地址：https://arxiv.org/abs/2211.09788

項目地址（剛剛開源）：

https://github.com/ShoufaChen/DiffusionDet

該研究發(fā)現(xiàn)在傳統(tǒng)的目標檢測里，存在一個缺陷，即它們依賴于一組固定的可學習查詢。然后研究者就在思考：是否存在一種簡單的方法甚至不需要可學習查詢就能進行目標檢測？

為了回答這一問題，本文提出了 DiffusionDet，該框架可以直接從一組隨機框中檢測目標，它將目標檢測制定為從噪聲框到目標框的去噪擴散過程。這種從 noise-to-box 的方法不需要啟發(fā)式的目標先驗，也不需要可學習查詢，這進一步簡化了目標候選，并推動了檢測 pipeline 的發(fā)展。

如下圖 1 所示，該研究認為 noise-to-box 范式類似于去噪擴散模型中的 noise-to-image 過程，后者是一類基于似然的模型，通過學習到的去噪模型逐步去除圖像中的噪聲來生成圖像。

DiffusionDet 通過擴散模型解決目標檢測任務，即將檢測看作圖像中 bounding box 位置 (中心坐標) 和大小 (寬度和高度) 空間上的生成任務。在訓練階段，將方差表（schedule）控制的高斯噪聲添加到 ground truth box，得到 noisy box。然后使用這些 noisy box 從主干編碼器（如 ResNet， Swin Transformer）的輸出特征圖中裁剪感興趣區(qū)域（RoI）。最后，將這些 RoI 特征發(fā)送到檢測解碼器，該解碼器被訓練用來預測沒有噪聲的 ground truth box。在推理階段，DiffusionDet 通過反轉(zhuǎn)學習到的擴散過程生成 bounding box，它將噪聲先驗分布調(diào)整到 bounding box 上的學習分布。

方法概述

由于擴散模型迭代地生成數(shù)據(jù)樣本，因此在推理階段需要多次運行模型 f_θ。但是，在每一個迭代步驟中，直接在原始圖像上應用 f_θ在計算上很困難。因此，研究者提出將整個模型分為兩部分，即圖像編碼器和檢測解碼器，前者只運行一次以從原始輸入圖像 x 中提取深度特征表示，后者以該深度特征為條件，從噪聲框 z_t 中逐步細化框預測。

圖像編碼器將原始圖像作為輸入，并為檢測解碼器提取其高級特征。研究者使用 ResNet 等卷積神經(jīng)網(wǎng)絡和 Swin 等基于 Transformer 的模型來實現(xiàn) DiffusionDet。與此同時，特征金字塔網(wǎng)絡用于為 ResNet 和 Swin 主干網(wǎng)絡生成多尺度特征圖。

檢測解碼器借鑒了 Sparse R-CNN，將一組 proposal 框作為輸入，從圖像編碼器生成的特征圖中裁剪 RoI 特征，并將它們發(fā)送到檢測頭以獲得框回歸和分類結(jié)果。此外，該檢測解碼器由 6 個級聯(lián)階段組成。

訓練

在訓練過程中，研究者首先構(gòu)建了從真值框到噪聲框的擴散過程，然后訓練模型來反轉(zhuǎn)這個過程。如下算法 1 提供了 DiffusionDet 訓練過程的偽代碼。

真值框填充。對于現(xiàn)代目標檢測基準，感興趣實例的數(shù)量通常因圖像而異。因此，研究者首先將一些額外的框填充到原始真值框，這樣所有的框被總計為一個固定的數(shù)字 N_train。他們探索了幾種填充策略，例如重復現(xiàn)有真值框、連接隨機框或圖像大小的框。

框損壞。研究者將高斯噪聲添加到填充的真值框。噪聲尺度由如下公式（1）中的 α_t 控制，它在不同的時間步 t 中采用單調(diào)遞減的余弦調(diào)度。

訓練損失。檢測解碼器將 N_train 損壞框作為輸入，預測 N_train 對類別分類和框坐標的預測。同時在 N_train 預測集上應用集預測損失（set prediction loss）。

推理

DiffusionDet 的推理過程是從噪聲到目標框的去噪采樣過程。從在高斯分布中采樣的框開始，該模型逐步細化其預測，具體如下算法 2 所示。

采樣步驟。在每個采樣步驟中，將上一個采樣步驟中的隨機框或估計框發(fā)送到檢測解碼器，以預測類別分類和框坐標。在獲得當前步驟的框后，采用 DDIM 來估計下一步驟的框。

框更新。為了使推理更好地與訓練保持一致，研究者提出了框更新策略，通過用隨機框替換非預期的框以使它們恢復。具體來說，他們首先過濾掉分數(shù)低于特定閾值的非預期的框，然后將剩余的框與從高斯分布中采樣的新隨機框連接起來。

一次解決（Once-for-all）。得益于隨機框設計，研究者可以使用任意數(shù)量的隨機框和采樣步驟來評估 DiffusionDet。作為比較，以往的方法在訓練和評估期間依賴于相同數(shù)量的處理框，并且檢測解碼器在前向傳遞中僅使用一次。

實驗結(jié)果

在實驗部分，研究者首先展示了 DiffusionDet 的 Once-for-all 屬性，然后將 DiffusionDet 與以往在 MS-COCO 和 LVIS 數(shù)據(jù)集上成熟的檢測器進行比較。

DiffusionDet 的主要特性在于對所有推理實例進行一次訓練。一旦模型經(jīng)過訓練，它就可以用于更改推理中框的數(shù)量和樣本步驟數(shù)，如下圖 4 所示。DiffusionDet 可以通過使用更多框或 / 和更多細化步驟來實現(xiàn)更高的準確度，但代價是延遲率更高。因此，研究者將單個 DiffusionDet 部署到多個場景中，并在不重新訓練網(wǎng)絡的情況下獲得所需的速度 - 準確率權(quán)衡。

研究者將 DiffusionDet 與以往在 MS-COCO 和 LVIS 數(shù)據(jù)集上的檢測器進行了比較，具體如下表 1 所示。他們首先將 DiffusionDet 的目標檢測性能與以往在 MS-COCO 上的檢測器進行了比較。結(jié)果顯示，沒有細化步驟的 DiffusionDet 使用 ResNet-50 主干網(wǎng)絡實現(xiàn)了 45.5 AP，以較大的優(yōu)勢超越了以往成熟的方法，如 Faster R-CNN、RetinaNet、DETR 和 Sparse R-CNN。并且當主干網(wǎng)絡的尺寸擴大時，DiffusionDet 顯示出穩(wěn)定的提升。

下表 2 中展示了在更具挑戰(zhàn)性的 LVIS 數(shù)據(jù)集上的結(jié)果，可以看到，DiffusionDet 使用更多的細化步驟可以獲得顯著的增益。

審核編輯 ：李倩