Water scarcity is a critical global issue, necessitating efficient water purification and desalination methods. Membrane separation methods are environmentally friendly and consume less energy, making them more economical compared to other desalination and purification methods. This survey explores the application of artificial intelligence (AI) to predict membrane behaviour in water purification and desalination processes. Various AI platforms, including machine learning (ML) and artificial neural networks (ANNs), were utilised to model water flux, predict fouling behaviour, simulate micropollutant dynamics and optimise operational parameters. Specifically, models such as convolutional neural networks (CNNs), recurrent neural networks (RNNs) and support vector machines (SVMs) have demonstrated superior predictive capabilities in these applications. This review studies recent advancements, emphasising the superior predictive capabilities of AI models compared to traditional methods. Key findings include the development of AI models for various membrane separation techniques and the integration of AI concepts such as ML and ANNs to simulate membrane fouling, water flux and micropollutant behaviour, aiming to enhance wastewater treatment and optimise treatment and desalination processes. In conclusion, this review summarised the applications of AI in predicting the behaviour of membranes as well as their strengths, weaknesses and future directions of AI in membranes for water purification and desalination processes.