Title
Predicate Correlation Learning for Scene Graph Generation
Abstract
For a typical Scene Graph Generation (SGG) method in image understanding, there usually exists a large gap in the performance of the predicates' head classes and tail classes. This phenomenon is mainly caused by the semantic overlap between different predicates as well as the long-tailed data distribution. In this paper, a Predicate Correlation Learning (PCL) method for SGG is proposed to address the above problems by taking the correlation between predicates into consideration. To measure the semantic overlap between highly correlated predicate classes, a Predicate Correlation Matrix (PCM) is defined to quantify the relationship between predicate pairs, which is dynamically updated to remove the matrix's long-tailed bias. In addition, PCM is integrated into a predicate correlation loss function (L-PC) to reduce discouraging gradients of unannotated classes. The proposed method is evaluated on several benchmarks, where the performance of the tail classes is significantly improved when built on existing methods.
Year
DOI
Venue
2022
10.1109/TIP.2022.3181511
IEEE TRANSACTIONS ON IMAGE PROCESSING
Keywords
DocType
Volume
Image understanding, scene graph generation, long-tailed bias, predicate correlation, semantic overlap
Journal
31
Issue
ISSN
Citations 
1
1057-7149
0
PageRank 
References 
Authors
0.34
0
6
Name
Order
Citations
PageRank
Leitian Tao100.34
Li Mi221.03
Nannan Li3146.60
Xianhang Cheng400.34
Yaosi Hu500.34
Zhenzhong Chen61244101.41