Impact of loss functions on the performance of a deep neural network designed to restore low-dose digital mammography

======

This repository contains the training and testing codes for the paper "Impact of loss functions on the performance of a deep neural network designed to restore low-dose digital mammography". For simulating dose reduction on clinical images, we used the codes available here. Also, we used a model-based (MB) restoration as a benchmark, also available here, which uses the commonly known BM3D.

Network architecture:

Some results:

Restoration of images with a dose reduction factor of 75%:

Restoration of images with a dose reduction factor of 50%:

Reference:

If you use the toolbox, we will be very grateful if you refer to this paper:

AI-based X-ray Imaging System (AXIS)
Department of Biomedical Engineering
Rensselaer Polytechnic Institute
Troy - USA

Laboratory of Computer Vision (Lavi)
Department of Electrical and Computer Engineering
São Carlos School of Engineering, University of São Paulo
São Carlos - Brazil

Low-dose Digital Mammography with Deep Learning

Related tags

Overview

Impact of loss functions on the performance of a deep neural network designed to restore low-dose digital mammography

Network architecture:

Some results:

Reference:

Owner

WANG-AXIS

WarpDrive: Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning on a GPU

A unified 3D Transformer Pipeline for visual synthesis

Deep Sea Treasure Environment for Multi-Objective Optimization Research

A geometric deep learning pipeline for predicting protein interface contacts.

1st Place Solution to ECCV-TAO-2020: Detect and Represent Any Object for Tracking

Pytorch implementation of SimSiam Architecture

This is the repository for the NeurIPS-21 paper [Contrastive Graph Poisson Networks: Semi-Supervised Learning with Extremely Limited Labels].

Code for "Intra-hour Photovoltaic Generation Forecasting based on Multi-source Data and Deep Learning Methods."

[IEEE Transactions on Computational Imaging] Self-Gated Memory Recurrent Network for Efficient Scalable HDR Deghosting

UV matrix decompostion using movielens dataset

Distilled coarse part of LoFTR adapted for compatibility with TensorRT and embedded divices

[ICCV'21] Learning Conditional Knowledge Distillation for Degraded-Reference Image Quality Assessment

Pytorch based library to rank predicted bounding boxes using text/image user's prompts.

Disentangled Cycle Consistency for Highly-realistic Virtual Try-On, CVPR 2021

An unofficial PyTorch implementation of a federated learning algorithm, FedAvg.

Repository for publicly available deep learning models developed in Rosetta community

An Exact Solver for Semi-supervised Minimum Sum-of-Squares Clustering

Deeplab-resnet-101 in Pytorch with Jaccard loss

Code for "The Box Size Confidence Bias Harms Your Object Detector"

High-Fidelity Pluralistic Image Completion with Transformers (ICCV 2021)