25–28 Sept 2023
Imperial College London
Europe/London timezone

AI Upscaling with Super Resolution CNNs on FPGAs and ASICs

27 Sept 2023, 17:25
5m
Blackett Laboratory, Lecture Theatre 1 (Imperial College London)

Blackett Laboratory, Lecture Theatre 1

Imperial College London

Blackett Laboratory
Lightning Talk Contributed Talks Contributed Talks

Speaker

Ryan Forelli (Lehigh University)

Description

Deep learning techniques have demonstrated remarkable performance in super resolution (SR) tasks for enhancing image resolution and granularity. These architectures extract image features with a convolutional block and add the extracted features to the upsampled input image transported through a skip connection, which is then converted from a depth to higher resolution space. However, SR can be computationally expensive due to large three-dimensional inputs with outputs many times larger while demanding low latency, making large scale implementation in commercial video streaming applications challenging. To address this issue, we explore the viability of SR deployment on-chip to FPGA and ASIC devices for low latency and low power inference. We train and optimize our model using a range of techniques, including quantization-aware training, batch normalization, heterogeneous quantization, and FIFO depth optimization to achieve an implementation which fits within our resource and accuracy constraints. Using the DIV2K diverse image dataset and supplying input images which are downscaled by a factor of three, we achieve >30 PSNR at >2b quantization. We use this initial FPGA implementation as a proof of concept for future ASIC implementations.

Authors

Ben Hawks (Fermi National Accelerator Lab) Giuseppe Di Guglielmo (Fermilab) Javier Campos Jovan Mitrevski (Fermi National Accelerator Lab. (US)) Jules Muhizi Nhan Tran (Fermi National Accelerator Lab. (US)) Ryan Forelli (Lehigh University)

Presentation materials