Deep Reinforcement Learning Optimizes Graphene Nanopores for Efficient Desalination

TL;DR

This paper introduces a deep reinforcement learning framework combined with CNNs to optimize graphene nanopores for water desalination, achieving higher water flux and maintaining ion rejection efficiency through efficient, online design.

Contribution

It presents a novel DRL-based method for nanopore optimization that reduces experimental costs and improves desalination performance compared to traditional approaches.

Findings

DRL-designed nanopores have higher water flux.

DRL-designed nanopores maintain high ion rejection.

Semi-oval shape with rough edges enhances performance.

Abstract

Two-dimensional nanomaterials, such as graphene, have been extensively studied because of their outstanding physical properties. Structure and geometry optimization of nanopores on such materials is beneficial for their performances in real-world engineering applications, like water desalination. However, the optimization process often involves very large number of experiments or simulations which are expensive and time-consuming. In this work, we propose a graphene nanopore optimization framework via the combination of deep reinforcement learning (DRL) and convolutional neural network (CNN) for efficient water desalination. The DRL agent controls the growth of nanopore by determining the atom to be removed at each timestep, while the CNN predicts the performance of nanoporus graphene for water desalination: the water flux and ion rejection at a certain external pressure. With the synchronous feedback from CNN-accelerated desalination performance prediction, our DRL agent can optimize the nanoporous graphene efficiently in an online manner. Molecular dynamics (MD) simulations on promising DRL-designed graphene nanopores show that they have higher water flux while maintaining rival ion rejection rate compared to the normal circular nanopores. Semi-oval shape with rough edges geometry of DRL-designed pores is found to be the key factor for their high water desalination performance. Ultimately, this study shows that DRL can be a powerful tool for material design.