DAGrid: Directed Accumulator Grid
- URL: http://arxiv.org/abs/2306.02589v1
- Date: Mon, 5 Jun 2023 04:33:32 GMT
- Title: DAGrid: Directed Accumulator Grid
- Authors: Hang Zhang, Renjiu Hu, Xiang Chen, Rongguang Wang, Jinwei Zhang, and
Jiahao Li
- Abstract summary: We present the Directed Accumulator Grid (DAGrid), which allows geometric-preserving filtering in neural networks.
We show DAGrid has realized a 70.8% reduction in network parameter size and a 96.8% decrease in FLOPs.
It has also achieved improvements of 4.4% and 8.2% in the average Dice score and Dice score of the left ventricular mass.
- Score: 13.188564605481544
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Recent research highlights that the Directed Accumulator (DA), through its
parametrization of geometric priors into neural networks, has notably improved
the performance of medical image recognition, particularly with small and
imbalanced datasets. However, DA's potential in pixel-wise dense predictions is
unexplored. To bridge this gap, we present the Directed Accumulator Grid
(DAGrid), which allows geometric-preserving filtering in neural networks, thus
broadening the scope of DA's applications to include pixel-level dense
prediction tasks. DAGrid utilizes homogeneous data types in conjunction with
designed sampling grids to construct geometrically transformed representations,
retaining intricate geometric information and promoting long-range information
propagation within the neural networks. Contrary to its symmetric counterpart,
grid sampling, which might lose information in the sampling process, DAGrid
aggregates all pixels, ensuring a comprehensive representation in the
transformed space. The parallelization of DAGrid on modern GPUs is facilitated
using CUDA programming, and also back propagation is enabled for deep neural
network training. Empirical results show DAGrid-enhanced neural networks excel
in supervised skin lesion segmentation and unsupervised cardiac image
registration. Specifically, the network incorporating DAGrid has realized a
70.8% reduction in network parameter size and a 96.8% decrease in FLOPs, while
concurrently improving the Dice score for skin lesion segmentation by 1.0%
compared to state-of-the-art transformers. Furthermore, it has achieved
improvements of 4.4% and 8.2% in the average Dice score and Dice score of the
left ventricular mass, respectively, indicating an increase in registration
accuracy for cardiac images. The source code is available at
https://github.com/tinymilky/DeDA.
Related papers
- Geometric Learning-Based Transformer Network for Estimation of
Segmentation Errors [1.376408511310322]
We propose an approach to identify and measure erroneous regions in the segmentation map.
Our method can estimate error at any point or node in a 3D mesh generated from a possibly erroneous volumetric segmentation map.
We have evaluated our network on a high-resolution micro-CT dataset of the human inner-ear bony labyrinth structure.
arXiv Detail & Related papers (2023-08-09T16:58:03Z) - EGRC-Net: Embedding-induced Graph Refinement Clustering Network [66.44293190793294]
We propose a novel graph clustering network called Embedding-Induced Graph Refinement Clustering Network (EGRC-Net)
EGRC-Net effectively utilizes the learned embedding to adaptively refine the initial graph and enhance the clustering performance.
Our proposed methods consistently outperform several state-of-the-art approaches.
arXiv Detail & Related papers (2022-11-19T09:08:43Z) - Low-Energy Convolutional Neural Networks (CNNs) using Hadamard Method [0.0]
Convolutional neural networks (CNNs) are a potential approach for object recognition and detection.
A new approach based on the Hadamard transformation as an alternative to the convolution operation is demonstrated.
The method is helpful for other computer vision tasks when the kernel size is smaller than the input image size.
arXiv Detail & Related papers (2022-09-06T21:36:57Z) - Parameter Convex Neural Networks [13.42851919291587]
We propose the exponential multilayer neural network (EMLP) which is convex with regard to the parameters of the neural network under some conditions.
For late experiments, we use the same architecture to make the exponential graph convolutional network (EGCN) and do the experiment on the graph classificaion dataset.
arXiv Detail & Related papers (2022-06-11T16:44:59Z) - Two-Stream Graph Convolutional Network for Intra-oral Scanner Image
Segmentation [133.02190910009384]
We propose a two-stream graph convolutional network (i.e., TSGCN) to handle inter-view confusion between different raw attributes.
Our TSGCN significantly outperforms state-of-the-art methods in 3D tooth (surface) segmentation.
arXiv Detail & Related papers (2022-04-19T10:41:09Z) - LHNN: Lattice Hypergraph Neural Network for VLSI Congestion Prediction [70.31656245793302]
lattice hypergraph (LH-graph) is a novel graph formulation for circuits.
LHNN constantly achieves more than 35% improvements compared with U-nets and Pix2Pix on the F1 score.
arXiv Detail & Related papers (2022-03-24T03:31:18Z) - TSGCNet: Discriminative Geometric Feature Learning with Two-Stream
GraphConvolutional Network for 3D Dental Model Segmentation [141.2690520327948]
We propose a two-stream graph convolutional network (TSGCNet) to learn multi-view information from different geometric attributes.
We evaluate our proposed TSGCNet on a real-patient dataset of dental models acquired by 3D intraoral scanners.
arXiv Detail & Related papers (2020-12-26T08:02:56Z) - 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) - Fast Mesh Data Augmentation via Chebyshev Polynomial of Spectral
filtering [5.594792814661452]
Deep neural networks have been recognized as one of the powerful learning techniques in computer vision and medical image analysis.
In practice, there is often insufficient training data available and augmentation is used to expand the dataset.
This study proposes two unbiased augmentation methods, Laplace-Beltrami eigenfunction Data Augmentation (LB-eigDA) and Chebyshev Data Augmentation (C-pDA)
arXiv Detail & Related papers (2020-10-06T15:18:26Z) - Collaborative Boundary-aware Context Encoding Networks for Error Map
Prediction [65.44752447868626]
We propose collaborative boundaryaware context encoding networks called AEP-Net for error prediction task.
Specifically, we propose a collaborative feature transformation branch for better feature fusion between images and masks, and precise localization of error regions.
The AEP-Net achieves an average DSC of 0.8358, 0.8164 for error prediction task, and shows a high Pearson correlation coefficient of 0.9873.
arXiv Detail & Related papers (2020-06-25T12:42:01Z)
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.