Cross-Modal Retrieval Training With Swapped Class Labels
Find Innovative SolutionsGenerate Solutions
Solution Overview
Problem
Existing cross-modal retrieval methods, particularly contrastive learning, suffer from quadratic sampling complexity and the assumption that instances in different pairs are automatically irrelevant, leading to inefficient training and incorrect pair separations.
Innovation Solution
A method involving self-supervised learning with swapped assignment of class labels (SwAMP) is used to train feature extractors for different modalities, reducing complexity to linear by predicting and swapping class labels, and using supervised cross-entropy loss to align features.
Engineering Contradictions & Design Principles
Engineering Contradiction Analysis
1Reliability
If contrastive learning is used to train feature extractors, then cross-modal retrieval can be performed, but the training complexity becomes quadratic in the number of training data points
Solution Approach 1:
The patent segments the training process into two independent modalities: image-based training and text-based training. Each modality is trained separately using its own feature extractor and loss function, avoiding the need to compute pairwise relationships across all data points. This segmentation reduces the training complexity from quadratic to linear while maintaining cross-modal retrieval capability through the shared embedding space.
2Ease of manufacture
If contrastive learning assumes instances in different pairs are automatically irrelevant, then training is simplified, but relevant instances across different pairs are incorrectly pushed apart
Solution Approach 1:
The patent introduces an intermediary mechanism: class labels serve as mediators that bridge instances across different pairs. By predicting class labels for each instance independently and then using these labels to guide the training objective, the system can identify relevant instances across different pairs without making incorrect assumptions. This intermediary approach maintains training simplicity while improving relevance detection accuracy.
3Productivity
If paired training data with minimal annotation is used, then data preparation effort is reduced, but the lack of semantic class labels limits the ability to learn accurate similarity metrics
Solution Approach 1:
The patent applies self-service by enabling the model to generate its own training signals through self-supervised learning. Class labels are predicted automatically by the model itself during training, eliminating the need for manual annotation. These self-generated labels then serve as supervision signals to learn accurate similarity metrics, thus maintaining data preparation efficiency while improving similarity metric accuracy.
Data Source
AI summary
Broadly speaking, the present application generally relates to a method for training a machine learning, ML, model to perform real world cross-modal retrieval problems, and to a computer-implemented method and apparatus for performing real world cross-modal retrieval problems such as including text-based video retrieval, sketch-based image retrieval, and image-text retrieval using a trained machine learning, ML, model.


