AI battery optimization