• Home
  • Search
  • Convolutional Neural Network Training with Distributed K-FAC
  • Open Access IconOpen Access
  • Cite Icon24
  • https://doi.org/10.1109/sc41405.2020.00098Copy DOI Icon

Convolutional Neural Network Training with Distributed K-FAC

  • Nov 1, 2020
  • J Gregory Pauloski +4 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Training neural networks with many processors can reduce time-to-solution; however, it is challenging to maintain convergence and efficiency at large scales. The Kroneckerfactored Approximate Curvature (K-FAC) was recently proposed as an approximation of the Fisher Information Matrix that can be used in natural gradient optimizers. We investigate here a scalable K-FAC design and its applicability in convolutional neural network (CNN) training at scale. We study optimization techniques such as layer-wise distribution strategies, inverse-free second-order gradient evaluation, and dynamic K-FAC update decoupling to reduce training time while preserving convergence. We use residual neural networks (ResNet) applied to the CIFAR10 and ImageNet-1k datasets to evaluate the correctness and scalability of our K-FAC gradient preconditioner. With ResNet-50 on the ImageNet-1k dataset, our distributed K-FAC implementation converges to the 75.9% MLPerf baseline in 18-25% less time than does the classic stochastic gradient descent (SGD) optimizer across scales on a GPU cluster.

Similar Papers
  • Conference Article
  • Citations26

Coarse-to-fine multi-task training of convolutional neural networks for automated information extraction from cancer pathology reports

  • Mar 01, 2018
  • Mohammed Alawad +2
  • Conference Article
  • Citations19

FPGA-based training of convolutional neural networks with a reduced precision floating-point library

  • Dec 01, 2017
  • Roberto Dicecco +2
  • PDF
  • Research Article
  • Citations10

INVESTIGATIONS ON THE POTENTIAL OF CONVOLUTIONAL NEURAL NETWORKS FOR VEHICLE CLASSIFICATION BASED ON RGB AND LIDAR DATA

  • May 30, 2017
  • ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences
  • R Niessner +2
  • Research Article
  • Citations174

Data augmentation using Generative Adversarial Networks (GANs) for GAN-based detection of Pneumonia and COVID-19 in chest X-ray images

  • Jan 01, 2021
  • Informatics in Medicine Unlocked
  • Saman Motamed +2
  • Conference Article
  • Citations9

Reconstruction combined training for convolutional neural networks on character recognition

  • Aug 01, 2015
  • Li Chen +4
  • Conference Article
  • Citations24

Batch Normalization Processor Design for Convolution Neural Network Training and Inference

  • May 01, 2021
  • Yu-Sheng Ting +2
  • Research Article

Deep Learning Based Circuit Breaker Non - Full - Phase Operation State Monitoring Method

  • Oct 01, 2019
  • IOP Conference Series: Materials Science and Engineering
  • Tao Tao
  • Research Article

НЕЙРОСЕТЕВАЯ СИСТЕМА РАСЧЕТА УРОЖАЙНОСТИ ПОДСОЛНЕЧНИКА

  • Jan 01, 2024
  • Herald of Technological University
  • A.S Katasev +3
  • Research Article
  • Citations137

Seismic response prediction method for building structures using convolutional neural network

  • Jan 30, 2020
  • Structural Control and Health Monitoring
  • Byung Kwan Oh +2
  • Research Article
  • Citations1

Manufacturing-driven AI: synthetic image generation for automated fiber orientation analysis in reinforced polymers

  • Nov 10, 2025
  • The International Journal of Advanced Manufacturing Technology
  • César García-Gascón +3
  • Research Article
  • Citations5

Beyond Traditional Biometrics: Harnessing Chest X-Ray Features for Robust Person Identification

  • Aug 04, 2024
  • Acta Informatica Pragensia
  • Farah Hazem +5
  • Research Article
  • Citations4

Glass forming ability prediction of bulk metallic glasses based on fused strategy

  • May 01, 2024
  • Transactions of Nonferrous Metals Society of China
  • Ting Zhang +2
  • PDF
  • Research Article
  • Citations7

Cross-Turbine Training of Convolutional Neural Networks for SCADA-Based Fault Detection in Wind Turbines

  • Nov 03, 2020
  • Annual Conference of the PHM Society
  • Markus Ulmer +3
  • PDF
  • Research Article
  • Citations11

Measuring inter-individual differences in behavioural types of gilthead seabreams in the laboratory using deep learning.

  • May 05, 2022
  • PeerJ
  • Marco Signaroli +7
  • Research Article

Application-aided wound identification based on a multi-class classification convolutional neural network

  • Jun 27, 2025
  • Cuadernos de Educación y Desarrollo
  • Edna De Freitas Gomes Ruas +10
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.