AI network optimization