Stratifying KLR algebras of affine ADE types