We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

eess.IV

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Electrical Engineering and Systems Science > Image and Video Processing

Title: HierAttn: Effectively Learn Representations from Stage Attention and Branch Attention for Skin Lesions Diagnosis

Abstract: An accurate and unbiased examination of skin lesions is critical for the early diagnosis and treatment of skin cancers. The visual feature of the skin lesions varies significantly because skin images are collected from patients with different skin colours by using various devices. Recent studies have developed ensembled convolutional neural networks (CNNs) to classify the images for early diagnosis. However, the practical use of CNNs is limited because their network structures are heavyweight and neglect contextual information. Vision transformers (ViTs) learn the global features by self-attention mechanisms, but they also have comparatively large model sizes (more than 100M). To address these limitations, we introduce HierAttn, a lite and effective neural network with hierarchical and self attention. HierAttn applies a novel strategy based on learning local and global features by a multi-stage and hierarchical network. The efficacy of HierAttn was evaluated by using the dermoscopy images dataset ISIC2019 and smartphone photos dataset PAD-UFES-20. The experimental results show that HierAttn achieves the best top-1 accuracy and AUC among state-of-the-art mobile networks, including MobileNetV3 and MobileViT. The code is available at this https URL
Comments: The code will be available at this https URL as soon as possible
Subjects: Image and Video Processing (eess.IV); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG)
Cite as: arXiv:2205.04326 [eess.IV]
  (or arXiv:2205.04326v1 [eess.IV] for this version)

Submission history

From: Wei Dai [view email]
[v1] Mon, 9 May 2022 14:30:34 GMT (1282kb)
[v2] Tue, 10 May 2022 06:49:58 GMT (3123kb,D)
[v3] Mon, 16 May 2022 09:37:08 GMT (3129kb,D)
[v4] Sun, 22 May 2022 04:44:32 GMT (3093kb,D)
[v5] Fri, 3 Jun 2022 02:24:46 GMT (8172kb,D)
[v6] Tue, 23 Aug 2022 05:50:01 GMT (2018kb,D)

Link back to: arXiv, form interface, contact.