Research & Papers

OSCS-SupCon: New Method Boosts Contrastive Learning Accuracy by 3.4%

Sigmoid-based loss and orthogonality constraints solve key limitations in feature disentanglement.

Deep Dive

Supervised Contrastive Learning (SupCon) has excelled in computer vision by modeling pairwise relationships among samples, but it suffers from two major flaws: negative-sample dilution from the standard InfoNCE loss, and entanglement of category-relevant (common) and irrelevant (style) features. To address this, Bin Wang and Fadi Dornaika introduce OSCS-SupCon (Orthogonal Sigmoid-based Common and Style Supervised Contrastive Learning). Their framework replaces the traditional InfoNCE loss with a sigmoid-based contrastive loss that includes two learnable parameters—temperature and bias—adaptively modulating pairwise decision boundaries to prevent dilution. Additionally, they enforce orthogonality between common and style feature subspaces via a linear projection with ReLU nonlinearity, drastically reducing feature overlap and improving disentanglement of style-irrelevant representations.

Extensive experiments on six benchmark datasets show OSCS-SupCon consistently outperforms state-of-the-art methods across multiple backbone architectures. Notably, on the fine-grained CUB200-2011 dataset with a ResNet-18 backbone, it achieves a 3.4% improvement in classification accuracy over the previous best method (CS-SupCon). Ablation studies confirm the effectiveness of each component—the sigmoid loss and orthogonality constraints—proving that disentangling common and style features directly boosts robustness and generalization. This work provides a practical drop-in improvement for any supervised contrastive learning pipeline, with implications for more reliable vision systems in domains like medical imaging and autonomous driving.

Key Points
  • Introduces a sigmoid-based contrastive loss with learnable temperature and bias to adaptively modulate decision boundaries and reduce negative-sample dilution.
  • Enforces explicit orthogonality between common and style feature subspaces using a linear projection with ReLU, improving feature disentanglement.
  • Achieves 3.4% higher classification accuracy on CUB200-2011 (ResNet-18) than state-of-the-art CS-SupCon, with consistent gains across six benchmarks.

Why It Matters

Better feature disentanglement leads to more robust AI vision models, improving real-world classification accuracy and generalization.

📬 Get the top 10 AI stories daily