Accurate and Robust Lesion RECIST Diameter Prediction and Segmentation
with Transformers
- URL: http://arxiv.org/abs/2208.13113v1
- Date: Sun, 28 Aug 2022 01:43:21 GMT
- Title: Accurate and Robust Lesion RECIST Diameter Prediction and Segmentation
with Transformers
- Authors: Youbao Tang, Ning Zhang, Yirui Wang, Shenghua He, Mei Han, Jing Xiao,
Ruei-Sung Lin
- Abstract summary: This paper proposes a transformer-based network for lesion RECIST diameter prediction and segmentation (LRDPS)
It is formulated as three correlative and complementary tasks: lesion segmentation, heatmap prediction, and keypoint regression.
MeaFormer achieves the state-of-the-art performance of LRDPS on the large-scale DeepLesion dataset.
- Score: 22.528235432455524
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Automatically measuring lesion/tumor size with RECIST (Response Evaluation
Criteria In Solid Tumors) diameters and segmentation is important for
computer-aided diagnosis. Although it has been studied in recent years, there
is still space to improve its accuracy and robustness, such as (1) enhancing
features by incorporating rich contextual information while keeping a high
spatial resolution and (2) involving new tasks and losses for joint
optimization. To reach this goal, this paper proposes a transformer-based
network (MeaFormer, Measurement transFormer) for lesion RECIST diameter
prediction and segmentation (LRDPS). It is formulated as three correlative and
complementary tasks: lesion segmentation, heatmap prediction, and keypoint
regression. To the best of our knowledge, it is the first time to use keypoint
regression for RECIST diameter prediction. MeaFormer can enhance
high-resolution features by employing transformers to capture their long-range
dependencies. Two consistency losses are introduced to explicitly build
relationships among these tasks for better optimization. Experiments show that
MeaFormer achieves the state-of-the-art performance of LRDPS on the large-scale
DeepLesion dataset and produces promising results of two downstream
clinic-relevant tasks, i.e., 3D lesion segmentation and RECIST assessment in
longitudinal studies.
Related papers
- Generative Principal Component Regression via Variational Inference [2.4415762506639944]
One approach to designing appropriate manipulations is to target key features of predictive models.
We develop a novel objective based on supervised variational autoencoders (SVAEs) that enforces such information is represented in the latent space.
We show in simulations that gPCR dramatically improves target selection in manipulation as compared to standard PCR and SVAEs.
arXiv Detail & Related papers (2024-09-03T22:38:55Z) - The effect of data augmentation and 3D-CNN depth on Alzheimer's Disease
detection [51.697248252191265]
This work summarizes and strictly observes best practices regarding data handling, experimental design, and model evaluation.
We focus on Alzheimer's Disease (AD) detection, which serves as a paradigmatic example of challenging problem in healthcare.
Within this framework, we train predictive 15 models, considering three different data augmentation strategies and five distinct 3D CNN architectures.
arXiv Detail & Related papers (2023-09-13T10:40:41Z) - Class Anchor Margin Loss for Content-Based Image Retrieval [97.81742911657497]
We propose a novel repeller-attractor loss that falls in the metric learning paradigm, yet directly optimize for the L2 metric without the need of generating pairs.
We evaluate the proposed objective in the context of few-shot and full-set training on the CBIR task, by using both convolutional and transformer architectures.
arXiv Detail & Related papers (2023-06-01T12:53:10Z) - FCN-Transformer Feature Fusion for Polyp Segmentation [12.62213319797323]
Colonoscopy is widely recognised as the gold standard procedure for the early detection of colorectal cancer.
The manual segmentation of polyps in colonoscopy images is time-consuming.
The use of deep learning for automation of polyp segmentation has become important.
arXiv Detail & Related papers (2022-08-17T15:31:06Z) - Fine-grained Correlation Loss for Regression [20.175415393263037]
We propose to revisit the classic regression tasks with novel investigations on directly optimizing the fine-grained correlation losses.
We extensively validate our method on two typical ultrasound image regression tasks, including the image quality assessment and bio-metric measurement.
arXiv Detail & Related papers (2022-07-01T11:25:50Z) - Cross-Site Severity Assessment of COVID-19 from CT Images via Domain
Adaptation [64.59521853145368]
Early and accurate severity assessment of Coronavirus disease 2019 (COVID-19) based on computed tomography (CT) images offers a great help to the estimation of intensive care unit event.
To augment the labeled data and improve the generalization ability of the classification model, it is necessary to aggregate data from multiple sites.
This task faces several challenges including class imbalance between mild and severe infections, domain distribution discrepancy between sites, and presence of heterogeneous features.
arXiv Detail & Related papers (2021-09-08T07:56:51Z) - Lesion Segmentation and RECIST Diameter Prediction via Click-driven
Attention and Dual-path Connection [16.80758525711538]
Measuring lesion size is an important step to assess tumor growth and monitor disease progression and therapy response.
We present a prior-guided dual-path network (PDNet) to segment common types of lesions throughout the whole body.
PDNet learns comprehensive and representative deep image features for our tasks and produces more accurate results.
arXiv Detail & Related papers (2021-05-05T02:00:14Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Deep Volumetric Universal Lesion Detection using Light-Weight Pseudo 3D
Convolution and Surface Point Regression [23.916776570010285]
Computer-aided lesion/significant-findings detection techniques are at the core of medical imaging.
We propose a novel deep anchor-free one-stage VULD framework that incorporates (1) P3DC operators to recycle the architectural configurations and pre-trained weights from the off-the-shelf 2D networks.
New SPR method to effectively regress the 3D lesion spatial extents by pinpointing their representative key points on lesion surfaces.
arXiv Detail & Related papers (2020-08-30T19:42:06Z) - DONet: Dual Objective Networks for Skin Lesion Segmentation [77.9806410198298]
We propose a simple yet effective framework, named Dual Objective Networks (DONet), to improve the skin lesion segmentation.
Our DONet adopts two symmetric decoders to produce different predictions for approaching different objectives.
To address the challenge of large variety of lesion scales and shapes in dermoscopic images, we additionally propose a recurrent context encoding module (RCEM)
arXiv Detail & Related papers (2020-08-19T06:02:46Z) - AutoHR: A Strong End-to-end Baseline for Remote Heart Rate Measurement
with Neural Searching [76.4844593082362]
We investigate the reason why existing end-to-end networks perform poorly in challenging conditions and establish a strong baseline for remote HR measurement with architecture search (NAS)
Comprehensive experiments are performed on three benchmark datasets on both intra-temporal and cross-dataset testing.
arXiv Detail & Related papers (2020-04-26T05:43:21Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.