Accelerate Literature Icon
Want to do a literature review? Try our new Literature Review workflow

The hybrid approach—convolutional neural networks and expectation maximisation algorithm—for tomographic reconstruction of hyperspectral images

  • Abstract
  • PDF
  • Literature Map
  • Similar Papers
Abstract
Translate article icon Translate Article Star icon

We present a simple, but novel, hybrid approach to hyperspectral data cube reconstruction from computed tomography imaging spectrometry (CTIS) images that sequentially combines neural networks and the iterative expectation maximisation (EM) algorithm. We train and test the ability of the method to reconstruct data cubes of 100 × 100 × 25 and 100 × 100 × 100 voxels, corresponding to 25 and 100 spectral channels, from simulated CTIS images generated by our CTIS simulator. The hybrid approach utilises the inherent strength of the Convolutional Neural Network (CNN) with regards to noise and its ability to yield consistent reconstructions and make use of the EM algorithm’s ability to generalise to spectral images of any object without training. The hybrid approach achieves better performance than both the CNNs and EM alone for seen (included in CNN training) and unseen (excluded from CNN training) cubes for both the 25- and 100-channel cases. For the 25 spectral channels, the improvements from CNN to the hybrid model (CNN + EM) in terms of the mean-squared errors are between 14 % and 26 %. For 100 spectral channels, the improvements between 19 % and 40 % are attained with the largest improvement of 40 % for the unseen data, to which the CNNs are not exposed during the training.

Similar Papers
  • Research Article
  • Cite Count Icon 3
  • 10.1155/2022/8387364
Acceleration of Deep Neural Network Training Using Field Programmable Gate Arrays.
  • Oct 17, 2022
  • Computational Intelligence and Neuroscience
  • Guta Tesema Tufa + 2 more

Convolutional neural network (CNN) training often necessitates a considerable amount of computational resources. In recent years, several studies have proposed for CNN inference and training accelerators in which the FPGAs have previously demonstrated good performance and energy efficiency. To speed up the processing, CNN requires additional computational resources such as memory bandwidth, a FPGA platform resource usage, time, power consumption, and large datasets for training. They are constrained by the requirement for improved hardware acceleration to support scalability beyond existing data and model sizes. This paper proposes a procedure for energy efficient CNN training in collaboration with an FPGA-based accelerator. We employed optimizations such as quantization, which is a common model compression technique, to speed up the CNN training process. Additionally, a gradient accumulation buffer is used to ensure maximum operating efficiency while maintaining gradient descent of the learning algorithm. To validate the design, we implemented the AlexNet and VGG-16 models on an FPGA board and laptop CPU along side GPU. It achieves 203.75 GOPS on Terasic DE1 SoC with the AlexNet model and 196.50 GOPS with the VGG-16 model on Terasic DE-SoC. Our result also exhibits that the FPGA accelerators are more energy efficient than other platforms.

  • Research Article
  • Cite Count Icon 18
  • 10.1109/tpami.2024.3377372
Real-Time CNN Training and Compression for Neural-Enhanced Adaptive Live Streaming.
  • Sep 1, 2024
  • IEEE transactions on pattern analysis and machine intelligence
  • Seunghwa Jeong + 7 more

We propose a real-time convolutional neural network (CNN) training and compression method for delivering high-quality live video even in a poor network environment. The server delivers a low-resolution video segment along with the corresponding CNN for super resolution (SR), after which the client applies the CNN to the segment in order to recover high-resolution video frames. To generate a trained CNN corresponding to a video segment in real-time, our method rapidly increases the training accuracy by promoting the overfitting property of the CNN while also using curriculum-based training. In addition, assuming that the pretrained CNN is already downloaded on the client side, we transfer only residual values between the updated and pretrained CNN parameters. These values can be quantized with low bits in real time while minimizing the amount of loss, as the distribution range is significantly narrower than that of the updated CNN. Quantitatively, our neural-enhanced adaptive live streaming pipeline (NEALS) achieves higher SR accuracy and a lower CNN compression loss rate within a constrained training time compared to the state-of-the-art CNN training and compression method. NEALS achieves 15 to 48% higher quality of the user experience compared to state-of-the-art neural-enhanced live streaming systems.

  • PDF Download Icon
  • Research Article
  • 10.15587/2706-5448.2022.252695
Comparative characteristics of the ability of convolutional neural networks to the concept of transfer learning
  • Feb 11, 2022
  • Technology audit and production reserves
  • Vladimir Khotsyanovsky

The object of research is the ability to combine a previously trained model of a deep neural network of direct propagation with user data when used in problems of determining the class of one object in the image. That is, the processes of transfer learning in convolutional neural networks in classification problems are considered. The conducted researches are based on application of a method of comparison of theoretical and practical results received at training of convolutional neural networks. The main objective of this research is to conduct two different learning processes. Traditional training during which in each epoch of training there is an adjustment of values of all weights of each layer of a network. After that there is a process of training of a neural network on a sample of the data presented by images. The second process is learning using transfer learning methods, when initializing a pre-trained network, the weights of all its layers are «frozen» except for the last fully connected layer. This layer will be replaced by a new one with the number of outputs, which should be equal to the number of classes in the sample. After that, to initialize its parameters by the random values distributed according to the normal law. Then conduct training of such convolutional neural network on the set sample. When the training was conducted, the results were compared. In conclusion, learning from convolutional neural networks using transfer learning techniques can be applied to a variety of classification tasks, ranging from numbers to space objects (stars and quasars). The amount of computer resources spent on research is also quite important. Because not all model of a convolutional neural network can be fully taught without powerful computer systems and a large number of images in the training sample.

  • Conference Article
  • Cite Count Icon 14
  • 10.1109/escience.2016.7870906
Detecting wildlife in uncontrolled outdoor video using convolutional neural networks
  • Oct 1, 2016
  • Connor Bowley + 3 more

This paper explores the use of Convolutional Neural Networks (CNNs) to detect Interior Least Tern in uncontrolled outdoor videos for the Wildlife@Home project. To be able to use CNNs on this video, this work developed strategies to bridge the gap between video collected by wildlife biologists and the methodlogies common for training and testing CNNs by utilizing a striding methodology to extract positive and negative training examples of a fixed size. Then in order to efficiently run trained CNNs over full videos, software was developed using OpenCL which was capable of utilizing multiple GPUs and other OpenCL capable compute devices concurrently. It was also shown that an already trained CNN can be further refined by training it further on new imagery, without having to retrain the whole network from scratch, saving significant time. Further, while the CNNs trained were only for detection of Interior Least Terns, they show promise for actually detecting behavior, as obvious peaks resulted for periods of video when a tern was in flight. To the authors' knowledge, this is the first attempt to utilize CNNs for the task of detecting wildlife in uncontrolled outdoor video.

  • PDF Download Icon
  • Research Article
  • Cite Count Icon 24
  • 10.1038/s41598-021-97195-6
Research on improved convolutional wavelet neural network
  • Sep 9, 2021
  • Scientific Reports
  • Jingwei Liu + 4 more

Artificial neural networks (ANN) which include deep learning neural networks (DNN) have problems such as the local minimal problem of Back propagation neural network (BPNN), the unstable problem of Radial basis function neural network (RBFNN) and the limited maximum precision problem of Convolutional neural network (CNN). Performance (training speed, precision, etc.) of BPNN, RBFNN and CNN are expected to be improved. Main works are as follows: Firstly, based on existing BPNN and RBFNN, Wavelet neural network (WNN) is implemented in order to get better performance for further improving CNN. WNN adopts the network structure of BPNN in order to get faster training speed. WNN adopts the wavelet function as an activation function, whose form is similar to the radial basis function of RBFNN, in order to solve the local minimum problem. Secondly, WNN-based Convolutional wavelet neural network (CWNN) method is proposed, in which the fully connected layers (FCL) of CNN is replaced by WNN. Thirdly, comparative simulations based on MNIST and CIFAR-10 datasets among the discussed methods of BPNN, RBFNN, CNN and CWNN are implemented and analyzed. Fourthly, the wavelet-based Convolutional Neural Network (WCNN) is proposed, where the wavelet transformation is adopted as the activation function in Convolutional Pool Neural Network (CPNN) of CNN. Fifthly, simulations based on CWNN are implemented and analyzed on the MNIST dataset. Effects are as follows: Firstly, WNN can solve the problems of BPNN and RBFNN and have better performance. Secondly, the proposed CWNN can reduce the mean square error and the error rate of CNN, which means CWNN has better maximum precision than CNN. Thirdly, the proposed WCNN can reduce the mean square error and the error rate of CWNN, which means WCNN has better maximum precision than CWNN.

  • Conference Article
  • Cite Count Icon 12
  • 10.1109/smartcloud.2016.31
FPGA-Based Accelerator for Deep Convolutional Neural Networks for the SPARK Environment
  • Nov 1, 2016
  • Raghid Morcel + 2 more

Deep Convolutional Neural Networks are known for their high performance. However, their complexity is one of their most challenging aspects. In this paper, we propose a design of an FPGA-based accelerator for the distributed training of convolutional neural networks. The accelerator is intended for use in the SPARK run time environment to accelerate the training of deep convolutional neural networks in the data center. Our accelerator is very energy efficient and achieves 40 to 250 times speedup in the computation of the multi-layer convolution operation, a key operation in the training/inference of deep networks.

  • Research Article
  • Cite Count Icon 137
  • 10.1002/stc.2519
Seismic response prediction method for building structures using convolutional neural network
  • Jan 30, 2020
  • Structural Control and Health Monitoring
  • Byung Kwan Oh + 2 more

In this study, a method of predicting the seismic responses of building structures based on a convolutional neural network (CNN) is proposed. In the method, the time histories of acceleration responses previously measured in a building during earthquakes are used in the CNN input layer, with the corresponding time histories of the displacement responses being used in the CNN output layer. The correlations between the features automatically extracted from the acceleration responses by the convolution and pooling operations in the various CNN layers and the displacement responses in the CNN output layer are established through CNN training. The trained CNN predicts the displacement responses for a future earthquake event using only the measured acceleration responses. To verify the validity of the proposed method, a numerical study on the ASCE benchmark model and an experimental study on a reinforced concrete frame structure are conducted. In the numerical study, the structural responses of the ASCE model subject to artificial earthquakes are utilized for CNN training, and the performance of the trained CNN is verified through seismic response prediction. In the experimental study, shaking table tests are performed for various earthquakes to measure the seismic responses of the reinforced concrete test structure. The seismic response prediction performance of the proposed method is examined using CNN trained with the measured seismic responses. Furthermore, the performances of CNNs trained with different numbers of datasets generated by the proposed data generation method based on data overlapping with the same data pool are discussed with related to the number of CNN training iteration number.

  • Research Article
  • Cite Count Icon 1
  • 10.1088/1742-6596/2062/1/012008
Comparative evaluation of performance and scalability of convolutional neural network implementations on a multisystem HPC architecture
  • Nov 1, 2021
  • Journal of Physics: Conference Series
  • Sunil Pandey + 2 more

The convolutional neural network training algorithm has been implemented for a central processing unit based high performance multisystem architecture machine. The multisystem or the multicomputer is a parallel machine model which is essentially an abstraction of distributed memory parallel machines. In actual practice, this model corresponds to high performance computing clusters. The proposed implementation of the convolutional neural network training algorithm is based on modeling the convolutional neural network as a computational pipeline. The various functions or tasks of the convolutional neural network pipeline have been mapped onto the multiple nodes of a central processing unit based high performance computing cluster for task parallelism. The pipeline implementation provides a first level performance gain through pipeline parallelism. Further performance gains are obtained by distributing the convolutional neural network training onto the different nodes of the compute cluster. The two gains are multiplicative. In this work, the authors have carried out a comparative evaluation of the computational performance and scalability of this pipeline implementation of the convolutional neural network training with a distributed neural network software program which is based on conventional multi-model training and makes use of a centralized server. The dataset considered for this work is the North Eastern University’s hot rolled steel strip surface defects imaging dataset. In both the cases, the convolutional neural networks have been trained to classify the different defects on hot rolled steel strips on the basis of the input image. One hundred images corresponding to each class of defects have been used for the training in order to keep the training times manageable. The hyperparameters of both the convolutional neural networks were kept identical and the programs were run on the same computational cluster to enable fair comparison. Both the convolutional neural network implementations have been observed to train to nearly 80% training accuracy in 200 epochs. In effect, therefore, the comparison is on the time taken to complete the training epochs.

  • Conference Article
  • 10.1109/metroagrifor50201.2020.9277632
Evaluation of Virtual Methods for Training Neural Networks in Agricultural Applications
  • Nov 4, 2020
  • Jorge Luis Jimenez Aparicio + 4 more

Improvement of productivity and efficiency in the agriculture sector points towards the necessity of developing autonomous vehicles. Here, the detection and classification of objects plays a major role, which can be achieved by using camera sensors and convolutional neural networks (CNNs). In order to train CNNs to perform correctly, good datasets are required. However, especially for the case of agriculture, datasets that contain relevant objects and are labeled (i.e. annotated with ground truth information) are not only scarce but also difficult to generate as this entails a high cost in resources and human labor. Therefore, we propose a different approach: using 3D simulation technology to generate relevant simulated sensor data which are implicitly labeled and a cost efficient solution to train neural networks. In this contribution, we assess the viability of training a CNN with simulated sensor data by comparing the achieved performance to a network trained with real sensor data. In addition, we evaluate the benefits of combining simulated data with real data for training CNNs, including complementary as well as Transfer Learning approaches. Finally, we show that using simulated sensor data for training CNNs is viable yet less accurate than using comparable real datasets and propose ways to improve simulations in this regard. To this end, we analyze various simulation factors in terms of their impact on the CNN performance and introduce further benefits of using simulated scenarios in general.

  • Conference Article
  • 10.1109/isbi.2019.8759488
Deep Network Anatomy Segmentation with Limited Annotations using Auxiliary Labels
  • Apr 1, 2019
  • Ahmed Harouni + 3 more

Deep convolutional neural networks (CNNs) have shown impressive performance in anatomy segmentation that are close to the state of the art atlas-based segmentation method. On one hand CNNs have 20x faster predictions than atlas-based segmentation. However, one of the main holdbacks of CNN’s advancement is that it’s training requires large amount of annotated data. This is a costly hurdle as annotation is time consuming and requires expensive medical expertise. The goal of this work is to reach state of the art segmentation performance using the minimum amount of expensive manual annotations. Recent studies show that auxiliary segmentations can be used together with manual annotations to improve CNN learning. To make this learning scheme more effective, we propose an image selection algorithm that wisely chooses images for manual annotation for producing more accurate auxiliary segmentations and a quality control algorithm that excludes poor quality auxiliary segmentations from CNN training. We perform extensive experiments over chest CT dataset by varying the number of manual annotations used for atlas-based methods and by varying the number of auxiliary segmentations to train the CNN. Our results show that CNN trained with auxiliary segmentations achieve higher dice of 0.76 vs 0.58 when trained with few accurate manual segmentations. Moreover, training with 100 or more auxiliary segmentations, the CNN always outperforms atlas-based method. Finally, when carefully selecting single atlas for producing auxiliary segmentations and controlling the quality of auxiliary segmentations, the trained CNN archives high average dice of 0.72 vs 0.62 when using a randomly selected image for manual annotation with all auxiliary segmentations.

  • PDF Download Icon
  • Research Article
  • Cite Count Icon 6
  • 10.1109/access.2021.3069882
Sparse Pixel Training of Convolutional Neural Networks for Land Cover Classification
  • Jan 1, 2021
  • IEEE Access
  • Noureldin Laban + 4 more

Convolutional Neural Networks (CNN) have become the core of modern machine learning approaches. In addition to its inspiring interior design idea, the success of CNN depends mainly on two factors, the first is the availability of training data and the second is the computing power of the used devices. In the field of remote sensing, data availability is difficult and expensive. Furthermore, processing large remote sensing data to accommodate different models is a laborious process. At the same time, training data is often collected in the form of points distributed over crop fields rather than regions, which results in the scarcity of training data. To specifically address the scarcity of training points, in this paper we present a sparse pixel-based training of U-Net convolutional neural networks for land cover classification. Training images are reconstructed from the points' collection in a random manner, they are used as an input for the convolution networks. Based on this proposed method, the amount of training data is reproduced from the different spectral signals for each land cover. We conducted extensive experiments on eight classes, using ground truth data collected from several locations in Fayoum Governorate, Egypt. The obtained results showed the superiority of the proposed method over other methods.

  • Conference Article
  • Cite Count Icon 23
  • 10.1145/3400302.3415643
FPGA-based low-batch training accelerator for modern CNNs featuring high bandwidth memory
  • Nov 2, 2020
  • Shreyas K Venkataramanaiah + 6 more

Training convolutional neural networks (CNNs) requires intensive computations as well as a large amount of storage and memory access. While low bandwidth off-chip memories in prior FPGA works have hindered the system-level performance, modern FPGAs offer high bandwidth memory (HBM2) that unlocks opportunities to improve the throughput/energy of FPGA-based CNN training. This paper presents a FPGA accelerator for CNN training which (1) uses HBM2 for efficient off-chip communication, and (2) supports various training operations (e.g. residual connections, stride-2 convolutions) for modern CNNs. We analyze the impact of HBM2 on CNN training workloads, provide a comprehensive comparison with DDR3, and present the strategies to efficiently use HBM2 features for enhanced CNN training performance. For training ResNet-20/VGG-like CNNs for CIFAR-10 dataset with low batch size of 2, the proposed CNN training accelerator on Intel Stratix-10 MX FPGA demonstrates 1.4/1.7X energy-efficiency improvement compared to Stratix-10 GX FPGA with DDR3 memory, and 4.5/9.7 X energy-efficiency improvement compared to Tesla V100 GPU.

  • Research Article
  • Cite Count Icon 4
  • 10.1016/j.phro.2024.100684
Impact of annotation imperfections and auto-curation for deep learning-based organ-at-risk segmentation
  • Oct 1, 2024
  • Physics and Imaging in Radiation Oncology
  • Victor I.J Strijbis + 3 more

Background and purposeSegmentation imperfections (noise) in radiotherapy organ-at-risk segmentation naturally arise from specialist experience and image quality. Using clinical contours can result in sub-optimal convolutional neural network (CNN) training and performance, but manual curation is costly. We address the impact of simulated and clinical segmentation noise on CNN parotid gland (PG) segmentation performance and provide proof-of-concept for an easily implemented auto-curation countermeasure. Methods and MaterialsThe impact of segmentation imperfections was investigated by simulating noise in clean, high-quality segmentations. Curation efficacy was tested by removing lowest-scoring Dice similarity coefficient (DSC) cases early during CNN training, both in simulated (5-fold) and clinical (10-fold) settings, using our full radiotherapy clinical cohort (RTCC; N = 1750 individual PGs). Statistical significance was assessed using Bonferroni-corrected Wilcoxon signed-rank tests. Curation efficacies were evaluated using DSC and mean surface distance (MSD) on in-distribution and out-of-distribution data and visual inspection. ResultsThe curation step correctly removed median(range) 98(90–100)% of corrupted segmentations and restored the majority (1.2 %/1.3 %) of DSC lost from training with 30 % corrupted segmentations. This effect was masked when using typical (non-curated) validation data. In RTCC, 20 % curation showed improved model generalizability which significantly improved out-of-distribution DSC and MSD (p < 1.0e-12, p < 1.0e-6). Improved consistency was observed in particularly the medial and anterior lobes. ConclusionsUp to 30% case removal, the curation benefit outweighed the training variance lost through curation. Considering the notable ease of implementation, high sensitivity in simulations and performance gains already at lower curation fractions, as a conservative middle ground, we recommend 15% curation of training cases when training CNNs using clinical PG contours.

  • Conference Article
  • Cite Count Icon 1
  • 10.1145/3627703.3650067
NeuroFlux: Memory-Efficient CNN Training Using Adaptive Local Learning
  • Apr 22, 2024
  • Dhananjay Saikumar + 1 more

Efficient on-device Convolutional Neural Network (CNN) training in resource-constrained mobile and edge environments is an open challenge. Backpropagation is the standard approach adopted, but it is GPU memory intensive due to its strong inter-layer dependencies that demand intermediate activations across the entire CNN model to be retained in GPU memory. This necessitates smaller batch sizes to make training possible within the available GPU memory budget, but in turn, results in substantially high and impractical training time. We introduce NeuroFlux, a novel CNN training system tailored for memory-constrained scenarios. We develop two novel opportunities: firstly, adaptive auxiliary networks that employ a variable number of filters to reduce GPU memory usage, and secondly, block-specific adaptive batch sizes, which not only cater to the GPU memory constraints but also accelerate the training process. NeuroFlux segments a CNN into blocks based on GPU memory usage and further attaches an auxiliary network to each layer in these blocks. This disrupts the typical layer dependencies under a new training paradigm - 'adaptive local learning'. Moreover, NeuroFlux adeptly caches intermediate activations, eliminating redundant forward passes over previously trained blocks, further accelerating the training process. The results are twofold when compared to Backpropagation: on various hardware platforms, NeuroFlux demonstrates training speed-ups of 2.3× to 6.1× under stringent GPU memory budgets, and NeuroFlux generates streamlined models that have 10.9× to 29.4× fewer parameters.

  • Research Article
  • Cite Count Icon 28
  • 10.1049/iet-cvi.2019.0506
Optimisation‐based training of evolutionary convolution neural network for visual classification applications
  • Jun 30, 2020
  • IET Computer Vision
  • Shanshan Tu + 7 more

Training of the convolution neural network (CNN) is a problem of global optimisation. This study proposed a hybrid modified particle swarm optimisation (MPSO) and conjugate gradient (CG) algorithm for efficient training of CNN. The training involves MPSO–CG to avoid trapping in local minima. Particularly, improvements in the MPSO by introducing a novel approach for control parameters, improved parameters updating criteria, a novel parameter in the velocity update equation, and fusion of the CG allows handling the issues in training CNN. In this study, the authors validate the proposed MPSO algorithm on three benchmark mathematical test functions and also compared with three different variants of the baseline particle swarm optimisation algorithm. Furthermore, the performance of the proposed MPSO–CG is also compared with other training algorithms focusing on the analysis of computational cost, convergence, and accuracy based on a standard problem specific to classification applications on CIFAR‐10 dataset and face and skin detection dataset.

Save Icon
Up Arrow
Open/Close
Setting-up Chat
Loading Interface