update AI for depth-pruning