Source-linked AI summary
Crossbar-Net: A Novel Convolutional Network for Kidney Tumor Segmentation in CT Images
Qian Yu, Yinghuan Shi, Jinquan Sun, Yang Gao, Yakang Dai, Jianbing Zhu
TL;DR
Kidney-tumor segmentation in CT is difficult when tumors resemble surrounding tissue or vary internally. Crossbar-Net combines directional patches with cascaded vertical and horizontal sub-models, and the paper reports successful applications beyond kidney tumors.
Problem
CT kidney-tumor segmentation is challenging because tumors may resemble surrounding tissues and have intensity differences within the tumor.
Source: §I · pp. 1–2Method
Crossbar-Net uses crossbar patches for local detail and directional context, then iteratively trains vertical and horizontal sub-models using mis-segmented regions.
Source: §I · pp. 2–3, A. Our Methodology · p. 4Results
Crossbar-Net’s left-ventricle segmentation DSC ranked second overall, with GridNet the most accurate; the conclusion also reports successful cardiac and breast-mass applications.
Source: G. Crossbar-Net for Cardiac Segmentation · p. 13, §V · p. 13Takeaways & Limitations
The reported cardiac and breast-mass applications support the authors’ claim of promising generalization beyond kidney-tumor segmentation.
Source: §V · p. 13
Abstract
from arXiv · showhide
Due to the irregular motion, similar appearance and diverse shape, accurate segmentation of kidney tumor in CT images is a difficult and challenging task. To this end, we present a novel automatic segmentation method, termed as Crossbar-Net, with the goal of accurate segmenting the kidney tumors. Firstly, considering that the traditional learning-based segmentation methods normally employ either whole images or squared patches as the training samples, we innovatively sample the orthogonal non-squared patches (namely crossbar patches), to fully cover the whole kidney tumors in either horizontal or vertical directions. These sampled crossbar patches could not only represent the detailed local information of kidney tumor as the traditional patches, but also describe the global appearance from either horizontal or vertical direction using contextual information. Secondly, with the obtained crossbar patches, we trained a convolutional neural network with two sub-models (i.e., horizontal sub-model and vertical sub-model) in a cascaded manner, to integrate the segmentation results from two directions (i.e., horizontal and vertical). This cascaded training strategy could effectively guarantee the consistency between sub-models, by feeding each other with the most difficult samples, for a better segmentation. In the experiment, we evaluate our method on a real CT kidney tumor dataset, collected from 94 different patients including 3,500 images. Compared with the state-of-the-art segmentation methods, the results demonstrate the superior results of our method on dice ratio score, true positive fraction, centroid distance and Hausdorff distance. Moreover, we have extended our crossbar-net to a different task: cardiac segmentation, showing the promising results for the better generalization.
I. INTRODUCTION
Kidney-tumor segmentation in CT is difficult because tumors can resemble surrounding tissue and vary internally, while manual delineation is undesirable at scale. Crossbar-Net combines directional context with local detail and cascaded training of vertical and horizontal sub-models.
I. INTRODUCTION
- Accurate segmentation supports tumor quantification and classification, while manual delineation is undesirable because it can be subjective and must cover many images.The paper describes automatic CT kidney-tumor segmentation as being in high demand.Source: §I · p. 1
- Existing methods can fail when tumors resemble surrounding tissue or have internally dissimilar intensity, leaving accurate segmentation an ongoing need.The introduction contrasts easier, high-contrast cases with tumors connected to or visually similar to surrounding tissue.Source: §I · p. 2
- Crossbar patches span tumors horizontally or vertically, combining local detail with contextual and symmetrical information that squared patches may miss.The authors note that simply enlarging square patches or using whole images may introduce irrelevant noise and obscure boundary details.Source: §I · p. 2
- Crossbar-Net trains vertical and horizontal sub-models in cascaded rounds, using each model’s difficult regions to provide complementary training to the other.The paper also describes basic sampling to select discriminative patches and balance classes, and covering re-sampling to support complementary help.Source: §I · p. 2
- The authors report promising generalization on cardiac and breast-mass segmentation, and kidney-tumor training and testing times of about 1 hour and 1.5 seconds on a regular GPU.Crossbar patches require no additional parameters to train.Source: §I · p. 3
II. RELATED WORK
Prior kidney-tumor methods use low-level features, image-level CNNs, or square-patch CNNs. Crossbar-Net differs by combining directional context and local detail with iterative complementarity between sub-models.
II. RELATED WORK
- Earlier kidney-tumor methods using energy minimization or hand-crafted features perform well in simpler cases, but are less assured when tumor appearance resembles surrounding tissue.Examples include region growing, active contours, texture features, and classifiers.Source: §II · p. 3
- Medical-image CNNs borrowed for kidney-tumor segmentation used squared patches or whole-image inputs, while the authors identify square-patch locality as a limitation for this task.The related work describes both patch-based and image-based alternatives.Source: §II · p. 3
- Unlike prior image-level or squared-patch methods, Crossbar-Net uses non-squared patches to capture local detail and vertical and horizontal context.This distinction is presented as the approach’s feature-representation difference.Source: §II · p. 3
- Unlike other cascaded approaches, Crossbar-Net combines vertical and horizontal sub-models that iteratively improve by complementing one another.The authors describe enhancing regions around misclassified pixels during later rounds.Source: §II · p. 4
III. METHOD
The method frames segmentation as pixel-wise classification and trains directional sub-models through iterative re-sampling. At test time, it combines the trained models’ predictions by majority vote.
A. Our Methodology
- The method converts segmentation into pixel-wise classification, predicting whether a patch belongs to the tumor or non-tumor class.The framework includes both training and testing stages.Source: A. Our Methodology · p. 4
- At each training round, the method identifies mis-segmented regions and re-samples them for the other directional sub-model, alongside patches from basic sampling.Training repeats until error converges or the maximum number of rounds is reached.Source: A. Our Methodology · p. 4
- For a new CT image, the trained sub-models from each round produce segmentations that are combined by majority vote.Source: A. Our Methodology · p. 4
B. Crossbar Patch Sampling Strategy
Crossbar patches combine local detail with horizontal or vertical context, while basic sampling concentrates examples near tumor boundaries and covering re-sampling targets mis-segmented regions. The two patch directions are processed by sub-models with experimentally selected architectures.
1) Basic Sampling Strategy:
- Basic sampling uses denser non-tumor patches near the tumor and sparser patches farther away, while sampling tumor patches uniformly.The strategy emphasizes the boundary region considered hard to segment and uses different sampling intervals across rounds.Source: 1) Basic Sampling Strategy · p. 5
2) Covering Re-sampling Strategy:
- Covering re-sampling spans a mis-segmented vertical region with horizontal patches centered on its left, middle, and right lines, producing roughly 40 patches per vertical patch.The process also samples every three pixels along each line to limit redundancy.Source: 2) Covering Re-sampling Strategy · p. 5
C. Sub-Model Architecture
- Both sub-models use eight convolutional layers, two max-pooling layers, and a softmax layer, with the vertical model taking a 100 × 20 patch as input.The vertical model uses non-square convolutional kernels; architecture settings were experimentally determined by inner cross-validation.Source: C. Sub-Model Architecture · p. 5
- The horizontal sub-model takes a 20 × 100 patch, and the architecture may differ across segmentation scenarios.The paper uses the same architecture for cardiac segmentation and a different one for breast mass segmentation.Source: C. Sub-Model Architecture · p. 5
D. Training Sub-Models
Crossbar-Net trains vertical and horizontal sub-models in rounds, using each model’s mis-segmented regions to generate patches for the other. Basic sampling controls sample distribution, while covering re-sampling emphasizes difficult regions; rounds stop at a maximum or when training error no longer decreases significantly.
D. Training Sub-Models
- Initial vertical and horizontal sub-models are trained on crossbar patches sampled with the basic strategy, including tumor patches and non-tumor patches centered on contours.Non-tumor sampling intervals are smaller near the tumor boundary than farther away.Source: D. Training Sub-Models · p. 5
- In later rounds, covering re-sampling targets mis-segmented regions, while basic sampling controls the amount and distribution of samples and avoids duplicate patches.The two sampling strategies are combined to generate new patches for the sub-models.Source: D. Training Sub-Models · p. 6
- Training repeats updates to the horizontal and vertical sub-models until the maximum round count is reached or their training errors no longer decrease significantly.Source: D. Training Sub-Models · p. 6
- Vertical and horizontal sub-models complement one another during cascaded training, with patches from mis-segmented regions used to emphasize those regions in subsequent rounds.The authors note that boundaries blurred in one direction may be sharp in the other.Source: D. Training Sub-Models · p. 6
- The authors describe self-improvement across rounds as potentially arising from newly sampled training data and increased emphasis on regions both sub-models misclassify.They characterize the expected improvement on those regions as analogous to AdaBoost’s weight-updating approach.Source: D. Training Sub-Models · p. 6
E. Testing
At test time, Crossbar-Net extracts crossbar patches for each pixel and sends them through vertical and horizontal sub-models over multiple rounds. The final pixel classification is determined by majority voting across the sub-model outputs.
- For each pixel in a new image, crossbar patches are extracted and passed to the trained vertical and horizontal sub-models in each round.Source: E. Testing · p. 6
- Each sub-model predicts whether the central pixel of its patch belongs to the tumor region.Source: E. Testing · p. 6
- The final segmentation is determined by majority voting across the 2T sub-models produced over T rounds.Source: E. Testing · p. 6
IV. EXPERIMENTAL RESULTS
The experiments evaluate Crossbar-Net on kidney tumor CT data and describe additional breast mass and cardiac datasets. The section outlines the evaluation sequence, including comparisons with baselines and applications to other segmentation tasks.
IV. EXPERIMENTAL RESULTS
- The experiments examine Crossbar-Net’s characteristics, compare it with baseline methods on kidney tumors, and apply it to cardiac and breast mass segmentation.Source: §IV · pp. 6–7
A. Data
- The kidney tumor dataset contains 3,500 CT slices from 94 subjects, with one tumor per slice and physician annotations.The data were divided by subject into training, validation, and testing sets of 50, 8, and 36 subjects.Source: A. Data · p. 7
- The breast mass experiments use 1,000 selected DDSM images, split into 600 training, 100 validation, and 300 testing images.Experienced radiologists annotated tumor boundaries, and image regions outside the image in extracted patches are filled with black.Source: A. Data · p. 7
- The cardiac dataset comprises 7,980 MRI slices from 33 subjects, with 20 subjects for training, 3 for validation, and 10 for testing.Ground-truth contours are provided for the left ventricle’s endocardium and epicardium.Source: A. Data · p. 7
B. Evaluation Criteria
The section defines overlap, coverage, and distance criteria and examines how Crossbar-Net’s patch design and iterative training affect segmentation. Its sub-models improve across rounds, and weighted voting of all sub-models and crossbar patches outperform the tested alternatives.
B. Evaluation Criteria
- DSC measures overlap, TPF measures correctly predicted tumor-pixel coverage, and CD and HD measure distances between predicted and manual segmentations.Higher DSC and TPF indicate better overlap and coverage; smaller CD and HD indicate closer segmentations.Source: B. Evaluation Criteria · p. 7
C. Implementation Details
- Experiments were repeated three times with randomly selected training, validation, and test subsets, and reported performance was averaged.Sub-models converged within 20 epochs on kidney tumors and 25 epochs on cardiac and DDSM data.Source: C. Implementation Details · p. 7
D. Characteristics of Crossbar-Net
- Separately trained sub-models improved as rounds increased but required 10 rounds to converge, compared with 3 rounds under the proposed joint training manner.The separate-training experiment excluded the other directional sub-model throughout.Source: D. Characteristics of Crossbar-Net · p. 8
- Vertical and horizontal sub-models corrected different tumor regions, and later sub-models generally performed better than earlier ones.Some boundary errors were shared across directions and persisted in early rounds.Source: D. Characteristics of Crossbar-Net · p. 8
- Voting across all sub-models outperformed combining only V3 and H3 by around 2% in DSC and TPF, with V3 and H3 weighted 1.5 and the others 1.As rounds increased, majority voting remained superior to combining only the last two sub-models.Source: D. Characteristics of Crossbar-Net · p. 9
- Squared patches produced much lower DSC and TPF than crossbar patches, and enlarging squared patches to 100 × 100 brought no significant improvement.The authors suggest larger squared patches may add irrelevant noise when distinguishing boundaries.Source: D. Characteristics of Crossbar-Net · p. 9
E. Comparison to Other Methods on Kidney Tumor
On kidney tumors, Crossbar-Net outperformed the compared methods across overlap, tumor-pixel coverage, and distance measures. Its results were also strong across 600 sampled tumors, while comparisons included hand-crafted, patch-based, and image-based methods.
E. Comparison to Other Methods on Kidney Tumor
- The comparison included hand-crafted methods, multi-scale 2D- and 3D-CNNs, FCN-G, U-Net, V-Net, and SegCaps.The hand-crafted baselines extracted the kidney area, calculated features manually, and then classified or segmented tumors.Source: E. Comparison to Other Methods on… · p. 10
- Testing used 600 randomly sampled images from one test set comprising 30 subjects.The authors used density estimation to depict DSC distributions across these tumors.Source: E. Comparison to Other Methods on… · p. 10
- On 600 sampled kidney tumors, Crossbar-Net achieved DSC above 0.9 for most tumors; the 2D-CNN and hand-crafted methods had many low-DSC cases.U-Net performed better than those methods and the multi-scale 3D-CNN, while SegCaps achieved the second-best performance; U-Net and SegCaps also had some zero-DSC cases.Source: E. Comparison to Other Methods on… · p. 10
- Crossbar-Net achieved higher DSC and TPF and smaller HD and CD than the other compared methods on kidney tumors.Table III reports average values of all four metrics across test sets.Source: E. Comparison to Other Methods on… · p. 10, Table III · p. 10
- Crossbar-Net, including six sub-models, trained in about 1 hour on one GPU and segmented a new patient in less than 1.5 seconds.The reported patient had about 35 slices; U-Net was very close to Crossbar-Net in training and testing time.Source: E. Comparison to Other Methods on… · p. 12
F. Crossbar-Net for Breast Mass Segmentation
The authors evaluated Crossbar-Net on breast-mass segmentation using DDSM. They report slightly better DSC than other methods and observe progressive improvement among sub-models, alongside greater computational cost than on kidney and cardiac datasets.
F. Crossbar-Net for Breast Mass Segmentation
- DDSM sub-models again showed self-improvement and mutual help, consistent with the characteristics observed in kidney experiments.The passage identifies the breast-mass results as confirming these characteristics.Source: F. Crossbar-Net for Breast Mass… · p. 12
- Crossbar-Net was slightly superior to the other methods in breast-mass segmentation on DDSM.The comparison reports DSC values for Crossbar-Net and several other methods.Source: F. Crossbar-Net for Breast Mass… · p. 12
- All six sub-models took about 6 hours to train and about 15 seconds to segment a new DDSM subject.The authors attribute the higher cost than on kidney and cardiac datasets to larger patches and a larger segmentation model.Source: F. Crossbar-Net for Breast Mass… · p. 12
G. Crossbar-Net for Cardiac Segmentation
On cardiac MRI, Crossbar-Net improved across sub-model rounds and showed competitive results against reported methods. Its MYO segmentation outperformed the comparisons, while its LV DSC ranked second overall and its distance metrics were best among them.
G. Crossbar-Net for Cardiac Segmentation
- LV sub-model performance improved gradually across rounds in the same direction.The reported experiment segmented LV and myocardium in cardiac MRI.Source: G. Crossbar-Net for Cardiac Segmentation · p. 12
- For LV segmentation, Crossbar-Net had the second-best DSC overall, while its HD and AD were superior to those of the other compared methods.GridNet had the most accurate LV DSC; for MYO segmentation, Crossbar-Net outperformed all other compared methods.Source: G. Crossbar-Net for Cardiac Segmentation · p. 13
- The authors report better performance on cardiac segmentation than on kidney tumors because the LV cavity and myocardium have more regular shapes.Source: G. Crossbar-Net for Cardiac Segmentation · p. 13
V. CONCLUSION
Crossbar-Net combines crossbar patches, complementary directional sub-models, and resampling of mis-segmented regions. Its reported applications and future direction define the scope of the conclusion.
V. CONCLUSION
- Crossbar-Net combines crossbar patches, basic and covering re-sampling, and cascaded training as its three main design innovations.The method is reported to handle kidney tumors with varied size, shape, contrast, and appearance.Source: §V · p. 13
- Basic and covering re-sampling enhance mis-segmented regions while preventing either sub-model from becoming over-emphasized on them.The directional sub-models can complement each other’s segmentation and improve themselves through re-sampling.Source: §V · p. 13
- Reported applications to cardiac and breast mass segmentation support the paper’s claim that Crossbar-Net has a wide range of application.The authors identify extending symmetric information beyond horizontal and vertical axes as future work.Source: §V · pp. 13–14
- Future work would extend symmetric information from horizontal and vertical axes to other axes.Source: §V · p. 14