arXiv · 2305.02927
Forward-Forward Contrastive Learning
Abstract
Medical image classification is one of the most important tasks for computer-aided diagnosis. Deep learning models, particularly convolutional neural networks, have been successfully used for disease classification from medical images, facilitated by automated feature learning. However, the diverse imaging modalities and clinical pathology make it challenging to construct generalized and robust classifications. Towards improving the model performance, we propose a novel pretraining approach, namely Forward Forward Contrastive Learning (FFCL), which leverages the Forward-Forward Algorithm in a contrastive learning framework--both locally and globally. Our experimental results on the chest X-ray dataset indicate that the proposed FFCL achieves superior performance (3.69% accuracy over ImageNet pretrained ResNet-18) over existing pretraining models in the pneumonia classification task. Moreover, extensive ablation experiments support the particular local and global contrastive pretraining design in FFCL.
Explore related subjects
Keep this discovery
Md. Atik Ahamed, Jin Chen, Abdullah-Al-Zubaer Imran. 2023-05-04. Forward-Forward Contrastive Learning. https://arxiv.org/abs/2305.02927
Cite the original work for its findings. Save a collection to share your selection of sources.