Royal Bank of Canada
SYSTEM AND METHOD FOR A CONVOLUTIONAL NEURAL NETWORK FOR MULTI-LABEL CLASSIFICATION WITH PARTIAL ANNOTATIONS

Abstract:

Effectively training machine learning systems with incomplete/partial labels is a practical, technical problem that solutions described herein attempt to overcome. In particular, an approach to modify loss functions on a proportionality basis is noted in some embodiments. In other embodiments, a graph neural network is provided to help identify correlations/causations as between categories. In another set of embodiments, a prediction approach is described to, based on originally provided labels, predict labels for unlabelled training samples such that the proportion of labelled labels relative to all labels is increased.