Multi-relation Message Passing for Multi-label Text Classification
A well-known challenge associated with the multi-label classification problem is modelling dependencies between labels. Most attempts at modelling label dependencies focus on co-occurrences, ignoring the valuable information that can be extracted by detecting label subsets that rarely occur together. For example, consider customer product reviews; a product probably would not simultaneously be tagged by both "recommended" (i.e., reviewer is happy and recommends the product) and "urgent" (i.e., the review suggests immediate action to remedy an unsatisfactory experience). Aside from the consideration of positive and negative dependencies, the direction of a relationship should also be considered. For a multi-label image classification problem, the "ship" and "sea" labels have an obvious dependency, but the presence of the former implies the latter much more strongly than the other way around. These examples motivate the modelling of multiple types of bi-directional relationships between labels. In this paper, we propose a novel method, entitled Multi-relation Message Passing (MrMP), for the multi-label classification problem. Experiments on benchmark multi-label text classification datasets show that the MrMP module yields similar or superior performance compared to state-of-the-art methods. The approach imposes only minor additional computational and memory overheads.
Code (1)
Tasks
Classificationimage-classificationImage ClassificationMulti-Label ClassificationMUlTI-LABEL-ClASSIFICATIONMulti-Label Image ClassificationMulti Label Text ClassificationMulti-Label Text ClassificationRelationtext-classificationText ClassificationSimilar Papers 제목 키워드 기반
Attentional Multilabel Learning over Graphs: A Message Passing Approach
We address a largely open problem of multilabel classification over graphs. Unlike traditional vector input, a graph has rich variable-size substructures which are related to the labels in some ways. We believe that unco…
General ClassificationGraph Neural NetworkMulti-Label LearningMLGIB: Multi-Label Graph Information Bottleneck for Expressive and Robust Message Passing
Graph Neural Networks (GNNs) suffer from over-squashing in deep message passing, where information from exponentially growing neighborhoods is compressed into fixed-dimensional representations. We show that this issue be…
Neural Message Passing for Multi-Relational Ordered and Recursive Hypergraphs
Message passing neural network (MPNN) has recently emerged as a successful framework by achieving state-of-the-art performances on many graph-based learning tasks. MPNN has also recently been extended to multi-relational…
Correlation-Aware Graph Convolutional Networks for Multi-Label Node Classification
Multi-label node classification is an important yet under-explored domain in graph mining as many real-world nodes belong to multiple categories rather than just a single one. Although a few efforts have been made by uti…
ClassificationGraph MiningNode ClassificationUnbiased Heterogeneous Scene Graph Generation with Relation-aware Message Passing Neural Network
Recent scene graph generation (SGG) frameworks have focused on learning complex relationships among multiple objects in an image. Thanks to the nature of the message passing neural network (MPNN) that models high-order i…
Graph GenerationRelationRepresentation LearningScene Graph Generation