HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

LaplaceNet: A Hybrid Graph-Energy Neural Network for Deep Semi-Supervised Classification

Philip Sellars Angelica I. Aviles-Rivero Carola-Bibiane Schönlieb

LaplaceNet: A Hybrid Graph-Energy Neural Network for Deep Semi-Supervised Classification

Abstract

Semi-supervised learning has received a lot of recent attention as it alleviates the need for large amounts of labelled data which can often be expensive, requires expert knowledge and be time consuming to collect. Recent developments in deep semi-supervised classification have reached unprecedented performance and the gap between supervised and semi-supervised learning is ever-decreasing. This improvement in performance has been based on the inclusion of numerous technical tricks, strong augmentation techniques and costly optimisation schemes with multi-term loss functions. We propose a new framework, LaplaceNet, for deep semi-supervised classification that has a greatly reduced model complexity. We utilise a hybrid approach where pseudolabels are produced by minimising the Laplacian energy on a graph. These pseudo-labels are then used to iteratively train a neural-network backbone. Our model outperforms state-of-the art methods for deep semi-supervised classification, over several benchmark datasets. Furthermore, we consider the application of strong-augmentations to neural networks theoretically and justify the use of a multi-sampling approach for semi-supervised learning. We demonstrate, through rigorous experimentation, that a multi-sampling augmentation approach improves generalisation and reduces the sensitivity of the network to augmentation.

Code Repositories

psellcam/LaplaceNet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-image-classification-on-cifarLaplaceNet (CNN-13)
Percentage error: 4.99±0.08
semi-supervised-image-classification-on-cifarLaplaceNet (WRN-28-2)
Percentage error: 4.35±0.10
semi-supervised-image-classification-on-cifar-2LaplaceNet (WRN-28-8)
Percentage error: 22.11± 0.23

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp