Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Consistency Learning via Decoding Path Augmentation for Transformers in Human Object Interaction Detection

About

Human-Object Interaction detection is a holistic visual recognition task that entails object detection as well as interaction classification. Previous works of HOI detection has been addressed by the various compositions of subset predictions, e.g., Image -> HO -> I, Image -> HI -> O. Recently, transformer based architecture for HOI has emerged, which directly predicts the HOI triplets in an end-to-end fashion (Image -> HOI). Motivated by various inference paths for HOI detection, we propose cross-path consistency learning (CPC), which is a novel end-to-end learning strategy to improve HOI detection for transformers by leveraging augmented decoding paths. CPC learning enforces all the possible predictions from permuted inference sequences to be consistent. This simple scheme makes the model learn consistent representations, thereby improving generalization without increasing model capacity. Our experiments demonstrate the effectiveness of our method, and we achieved significant improvement on V-COCO and HICO-DET compared to the baseline models. Our code is available at https://github.com/mlvlab/CPChoi.

Jihwan Park, SeungJun Lee, Hwan Heo, Hyeong Kyu Choi, Hyunwoo J.Kim• 2022

Related benchmarks

TaskDatasetResultRank
Human-Object Interaction DetectionHICO-DET (test)
mAP (full)29.63
493
Human-Object Interaction DetectionV-COCO (test)
AP (Role, Scenario 1)63.1
270
Human-Object Interaction DetectionHICO-DET--
233
Human-Object Interaction DetectionV-COCO
AP^1 Role63.1
65
HOI DetectionV-COCO
AP Role 163.1
40
HOI DetectionHICO-DET
mAP (Default Full)29.63
21
Showing 6 of 6 rows

Other info

Code

Follow for update