Timezone: »
Poster
$\alpha$-IoU: A Family of Power Intersection over Union Losses for Bounding Box Regression
JIABO HE · Sarah Erfani · Xingjun Ma · James Bailey · Ying Chi · Xian-Sheng Hua
Bounding box (bbox) regression is a fundamental task in computer vision. So far, the most commonly used loss functions for bbox regression are the Intersection over Union (IoU) loss and its variants. In this paper, we generalize existing IoU-based losses to a new family of power IoU losses that have a power IoU term and an additional power regularization term with a single power parameter $\alpha$. We call this new family of losses the $\alpha$-IoU losses and analyze properties such as order preservingness and loss/gradient reweighting. Experiments on multiple object detection benchmarks and models demonstrate that $\alpha$-IoU losses, 1) can surpass existing IoU-based losses by a noticeable performance margin; 2) offer detectors more flexibility in achieving different levels of bbox regression accuracy by modulating $\alpha$; and 3) are more robust to small datasets and noisy bboxes.
Author Information
JIABO HE (the University of Melbourne)
Sarah Erfani (University of Melbourne)
Xingjun Ma (Deakin University)
James Bailey (THE UNIVERSITY OF MELBOURNE)
Ying Chi (Alibaba DAMO Academy)
Xian-Sheng Hua (Alibaba Cloud)
Researcher/Senior Director of Alibaba Cloud, in charge of visual computing team in Big Data Department of Alibaba Cloud. He is an IEEE Fellow, ACM Distinguished Scientist and a TR35 Young Innovator Award Recipient, for his outstanding contributions in image/video analysis and search area. Joined Alibaba in April of 2015. Before that, he worked for Microsoft Research, Microsoft Bing and Microsoft Research Asia for 14 years. More details please visit: https://www.linkedin.com/in/xshua
More from the Same Authors
-
2022 Poster: CalFAT: Calibrated Federated Adversarial Training with Label Skewness »
Chen Chen · Yuchen Liu · Xingjun Ma · Lingjuan Lyu -
2023 Poster: IDEA: An Invariant Perspective for Efficient Domain Adaptive Image Retrieval »
Haixin Wang · Hao Wu · Jinan Sun · Shikun Zhang · Chong Chen · Xian-Sheng Hua · Xiao Luo -
2022 Poster: Double Bubble, Toil and Trouble: Enhancing Certified Robustness through Transitivity »
Andrew Cullen · Paul Montague · Shijie Liu · Sarah Erfani · Benjamin Rubinstein -
2021 Poster: Gradient Driven Rewards to Guarantee Fairness in Collaborative Machine Learning »
Xinyi Xu · Lingjuan Lyu · Xingjun Ma · Chenglin Miao · Chuan Sheng Foo · Bryan Kian Hsiang Low -
2021 Poster: Anti-Backdoor Learning: Training Clean Models on Poisoned Data »
Yige Li · Xixiang Lyu · Nodens Koren · Lingjuan Lyu · Bo Li · Xingjun Ma -
2021 Poster: Exploring Architectural Ingredients of Adversarially Robust Deep Neural Networks »
Hanxun Huang · Yisen Wang · Sarah Erfani · Quanquan Gu · James Bailey · Xingjun Ma -
2020 Poster: AdvFlow: Inconspicuous Black-box Adversarial Attacks using Normalizing Flows »
Hadi Mohaghegh Dolatabadi · Sarah Erfani · Christopher Leckie