A low-cost edge AI-chip-based system for real-time algae species classification and HAB prediction

Harmful Algal Blooms (HAB) are damaging to ecosystem functions and pose challenges to environmental and fisheries management. The key to HAB management and understanding the complex algal growth dynamics is the development of robust systems for real-time monitoring of algae populations and species. Previous algae classification studies mainly rely on the combination of an in-situ imaging flow cytometer and an off-site lab-based algae classification model such as Random Forest (RF) for the analysis of high-throughput images. An on-site AI algae monitoring system on top of an edge AI chip embedded with the proposed Algal Morphology Deep Neural Network (AMDNN) model is developed to achieve real-time algae species classification and HAB prediction. Based on a detailed examination of real-world algae images, dataset augmentation is first performed: consisting of orientation, flipping, blurring, and Resizing with Aspect ratio Preserved (RAP). The dataset augmentation is shown to significantly improve classification performance which is superior to that of the competitive RF model. And the attention heatmaps show that for relatively regular-shaped algal species (e.g., Vicicitus), the model weights the color and texture information heavily; while the shape-related features are more important for complex-shaped algae (e.g., Chaetoceros). The AMDNN is tested on a dataset of 11,250 algae images containing the 25 most common HAB classes in Hong Kong subtropical waters with 99.87% test accuracy. Based on the fast and accurate algae classification, the AI-chip-based on-site system is applied to a one-month dataset in February 2020; the predicted trends of total cell counts and targeted HAB species counts are in good agreement with observations. The proposed edge AI algae monitoring system provides a platform for the development of practical HAB early warning systems that can effectively support environmental risk and fisheries management.