You can deploy this model and start earning money today!

soumickmj/
GPUNet_BraTS2020T1ce_Axial

Name: GPUNet_BraTS2020T1ce_Axial
Availability: InStock

Views

No views yet

GPUNet — BraTS 2020 · T1CE (single contrast) · Axial

Architecture — GPUNet

GPUNet is a U-Net variant with:

Hyperparameter	Value
Encoder/decoder depth	3
Base feature maps (`wf`)	6 → 2⁶ = 64 filters at level 0
Upsampling	sinc interpolation
Activation	ReLU
Dropout	✓ (p = 0.5)
Batch normalisation	✗
Padding	✓ (same)

The skip-connections preserve spatial detail, making the intermediate feature maps directly usable as segmentation-quality attention maps without extra supervision.

Overview

This model is part of the GPModels family — a set of inherently-explainable convolutional networks for simultaneous brain-tumour classification and weakly-supervised segmentation from multi-contrast MRI. The models were trained and evaluated on the BraTS 2020 dataset.

⚠️ Single-contrast model — not used in the paper

This model was trained on T1CE contrast only (1 input channel) as an exploratory experiment. It is not part of the original publication: Weakly-supervised segmentation using inherently-explainable classification models and their application to brain tumour classification, which exclusively uses 4-contrast (T1 · T2 · T1CE · FLAIR) models.

If you want a quick single-contrast experiment, this model may be useful.

If you want to reproduce the paper results, or need the best performance, please use the 4-contrast counterpart: GPUNet — 4-contrast

Input contrast: T1CE only (1 channel)
Task: Multi-class brain-tumour classification (Healthy / LGG / HGG) and weakly-supervised segmentation
Orientation: Axial 2-D slices (240 × 240 px)
Output classes: 0 = Healthy, 1 = LGG (Low-grade glioma), 2 = HGG (High-grade glioma)
Normalization: Per-image max normalization (divide by slice maximum)
Preprint (open access): https://arxiv.org/abs/2206.05148
Code & training details: https://github.com/soumickmj/GPModels

Model Inputs / Outputs

Property	Value
Input shape	`(B, 1, 240, 240)` — float32, max-normalized, T1CE only
Output — train mode	`(B, 3)` — raw logits
Output — eval mode	`((B, 3), (B, 3, H, W))` — logits and spatial heatmap
Class order	`[Healthy (0), LGG (1), HGG (2)]`

Usage

python

import torch
from transformers import AutoModel

model = AutoModel.from_pretrained("soumickmj/GPUNet_BraTS2020T1ce_Axial", trust_remote_code=True)
model.eval()

# x: (B, 1, 240, 240) — T1CE slice, max-normalised
x = torch.randn(1, 1, 240, 240)
with torch.no_grad():
    logits, heatmap = model(x)     # eval mode: logits (B,3)  +  heatmap (B,3,H,W)

pred_class = logits.argmax(dim=1)  # 0=Healthy, 1=LGG, 2=HGG

Weakly-Supervised Segmentation (Heatmap Mode)

These models are inherently explainable: eval mode bypasses the global max-pooling (GMP) layer, exposing a full-resolution spatial activation map per class. No extra labels or re-training are required.

How it works:

model.train() → GMP applied → returns logits (B, 3) — classification only
model.eval() → GMP skipped → returns (logits, heatmap) where heatmap has shape (B, 3, H, W) — one spatial map per class

Heatmap channel order: 0 = Healthy, 1 = LGG, 2 = HGG. The whole-tumour map is obtained by combining channels 1 and 2.

python

with torch.no_grad():
    logits, heatmap = model(x)             # heatmap: (B, 3, H, W)

# Whole-tumour map (max over LGG + HGG channels)
wt_map  = heatmap[:, 1:, :, :].max(dim=1).values   # (B, H, W)

# Min-max normalise to [0, 1]
wt_flat = wt_map.view(wt_map.size(0), -1)
wt_min  = wt_flat.min(dim=1).values[:, None, None]
wt_max  = wt_flat.max(dim=1).values[:, None, None]
wt_norm = (wt_map - wt_min) / (wt_max - wt_min + 1e-8)

# Binary mask via threshold
binary_mask = (wt_norm > 0.5).float()               # (B, H, W)

For advanced post-processing (multi-Otsu, top-k binarisation, morphological clean-up, per-slice aggregation) see the project repository.

Training Details

Setting	Value
Trained by	Soumick Chatterjee (Pavan)
Dataset	BraTS 2020 — non-empty axial slices
Input contrast	T1CE only (1 channel)
Split	Stratified 75/25 train-test (seed 13); 5-fold CV (fold 0 reported)
Optimizer	Adam (lr = 1e-3, weight_decay = 5e-4)
Loss	Cross-entropy with class-balanced weights
Mixed precision	AMP
Max epochs	300

Citation

If you use this model, please cite:

bibtex

@article{chatterjee2026weakly,
  title={Weakly-supervised segmentation using inherently-explainable classification models and their application to brain tumour classification},
  author={Chatterjee, Soumick and Yassin, Hadya and Dubost, Florian and N{\"u}rnberger, Andreas and Speck, Oliver},
  journal={Neurocomputing},
  pages={133460},
  year={2026},
  publisher={Elsevier}
}

License

MIT — see https://github.com/soumickmj/GPModels for full details.