Journal of Artificial Intelligence, Applications and Innovations 3060-7124 1 Journal of Artificial Intelligence, Application and Inovations 2024 10 01 A Novel U-Net Architecture with Attention Mechanism for Image Denoising A Novel U-Net Architecture with Attention Mechanism for Image Denoising 30 40 10.61838/jaiai.1.4.3 EN Journal Article 2024 06 01 <p>In this study, we present an enhanced U-Net-based model for effective image denoising, incorporating a hybrid attention mechanism that combines both spatial and channel attention. These dual attention blocks enable the network to dynamically focus on relevant features while suppressing noise across both dimensions, thereby improving denoising performance. To further refine the output and enhance perceptual quality, a Gaussian filter is applied as a post-processing step, resulting in smoother edges and better texture continuity. The model also leverages Batch Normalization and Dropout techniques to stabilize training and prevent overfitting. Experimental evaluations were conducted on the CIFAR-10 and DIV2K datasets using standard performance metrics. The proposed model achieved an accuracy of 82%, a loss of 0.01, a PSNR of 37 dB, and an SSIM of 0.94—outperforming several state-of-the-art denoising methods. These results confirm the model’s strong ability to preserve structural and textural image details while significantly reducing noise. The combination of convolutional deep learning, hybrid attention mechanisms, and post-processing filtering offers a powerful and scalable solution for image restoration tasks. Furthermore, it demonstrates strong potential for practical applications in real-world scenarios such as image quality enhancement and medical imaging.</p> https://www.journalaiai.com/index.php/aiai/article/download/63/25