Marginalized Generalized IoU (MGIoU): A Unified Objective Function for Optimizing Convex Parametric Shapes
Duy-Tho Le, Trung Pham, Jianfei Cai, Hamid Rezatofighi
Abstract
Optimizing the similarity between parametric shapes is crucial for numerous computer vision tasks, where Intersection over Union (IoU) stands as the canonical measure. However, existing optimization methods exhibit significant shortcomings: regression-based losses like L1/L2 lack correlation with IoU, IoU-based losses are unstable and limited to simple shapes, and task-specific methods are computationally intensive and not generalizable across domains. As a result, the current landscape of parametric shape objective functions has become scattered, with each domain proposing distinct IoU approximations. To address this, we unify the parametric shape optimization objective functions by introducing Marginalized Generalized IoU (MGIoU), a novel loss function that overcomes these challenges by projecting structured convex shapes onto their unique shape Normals to compute one-dimensional normalized GIoU. MGIoU offers a simple, efficient, fully differentiable approximation strongly correlated with IoU. We extend MGIoU to MGIoU+ that supports optimizing unstructured convex shapes. Together, MGIoU and MGIoU+ unify parametric shape optimization across diverse applications. Experiments on standard benchmarks demonstrate that MGIoU and MGIoU+ demonstrate higher performance while reducing loss computation latency up to 10-40x. Also, MGIoU and MGIoU+ satisfy metric properties and scale-invariance, ensuring robustness as an objective function. We further propose MGIoU- for minimizing overlaps in tasks like collision-free trajectory prediction.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ef2f8c22-6674-404a-9d22-3fb724100d6dCited by top-tier papers1
Ask how each one uses itBuilds on13
- Distance-IoU Loss: Faster and Better Learning for Bounding Box RegressionZhaohui Zheng, Ping Wang, Wei Liu, Jinze Li et al.AAAI 2020 · 4,823 citations
- Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene UnderstandingMike Roberts, Jason Ramapuram, Anurag Ranjan, Atulit Kumar et al.ICCV 2021 · 633 citations
- Learning High-Precision Bounding Box for Rotated Object Detection via Kullback-Leibler DivergenceXue Yang, Xiaojiang Yang, Jirui Yang, Qi Ming et al.NeurIPS 2021 · 603 citations
- Rethinking Rotated Object Detection with Gaussian Wasserstein Distance LossXue Yang, Junchi Yan, Qi Ming, Wentao Wang et al.ICML 2021 · 572 citations
- Motion Transformer with Global Intention Localization and Local Movement RefinementShaoshuai Shi, Li Jiang, Dengxin Dai, Bernt SchieleNeurIPS 2022 · 515 citations
Related papers
- Learning Generalized Intersection Over Union for Dense Pixelwise PredictionJiaqian Yu, Jingtao Xu, Yiwei Chen, Weiming Li et al.ICML 2021 · 16 citations
- Deep Dive into Gradients: Better Optimization for 3D Object Detection with Gradient-Corrected IoU SupervisionQi Ming, Lingjuan Miao, Zhe Ma, Lin Zhao et al.CVPR 2023
- -IoU: A Family of Power Intersection over Union Losses for Bounding Box RegressionJiabo He, Sarah M. Erfani, Xingjun Ma, James Bailey et al.NeurIPS 2021 · 334 citations
- SCALoss: Side and Corner Aligned Loss for Bounding Box RegressionTu Zheng, Shuai Zhao, Yang Liu, Zili Liu et al.AAAI 2022 · 14 citations
- Unbiased IoU for Spherical Image Object DetectionFeng Dai, Bin Chen, Hang Xu, Yike Ma et al.AAAI 2022 · 16 citations
