HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Learning Semantics-enriched Representation via Self-discovery, Self-classification, and Self-restoration

Fatemeh Haghighi; Mohammad Reza Hosseinzadeh Taher; Zongwei Zhou; Michael B. Gotway; Jianming Liang

Learning Semantics-enriched Representation via Self-discovery, Self-classification, and Self-restoration

Abstract

Medical images are naturally associated with rich semantics about the human anatomy, reflected in an abundance of recurring anatomical patterns, offering unique potential to foster deep semantic representation learning and yield semantically more powerful models for different medical applications. But how exactly such strong yet free semantics embedded in medical images can be harnessed for self-supervised learning remains largely unexplored. To this end, we train deep models to learn semantically enriched visual representation by self-discovery, self-classification, and self-restoration of the anatomy underneath medical images, resulting in a semantics-enriched, general-purpose, pre-trained 3D model, named Semantic Genesis. We examine our Semantic Genesis with all the publicly-available pre-trained models, by either self-supervision or fully supervision, on the six distinct target tasks, covering both classification and segmentation in various medical modalities (i.e.,CT, MRI, and X-ray). Our extensive experiments demonstrate that Semantic Genesis significantly exceeds all of its 3D counterparts as well as the de facto ImageNet-based transfer learning in 2D. This performance is attributed to our novel self-supervised learning framework, encouraging deep models to learn compelling semantic representation from abundant anatomical patterns resulting from consistent anatomies embedded in medical images. Code and pre-trained Semantic Genesis are available at https://github.com/JLiangLab/SemanticGenesis .

Code Repositories

fhaghighi/SemanticGenesis
pytorch
Mentioned in GitHub
JLiangLab/SemanticGenesis
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
brain-tumor-segmentation-on-brats-2013Semantic Genesis
Dice Score: 92.76
brain-tumor-segmentation-on-brats-2018Semantic Genesis
IoU: 68.8
liver-segmentation-on-lits2017Semantic Genesis
Dice: 92.27
IoU: 85.6
lung-nodule-detection-on-luna2016-fpredSemantic Genesis
AUC: 98.47
lung-nodule-segmentation-on-lidc-idriSemantic Genesis
IoU: 77.24

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
Learning Semantics-enriched Representation via Self-discovery, Self-classification, and Self-restoration | Papers | HyperAI