[D] Undergrad ML PhD wannabe: big lab or small lab?

I'm a 3rd year at a T5 (or T10, idk how rankings work) school, and I've been doing research since high school. I've got a few publications under my belt and my professor (at a different T3 institution) is expecting my current work will get me a first authorship to Nature Medicine (or a similar top journal). It's all been applied ML/DL work, especially in reducing the cost of healthcare. I've been keeping up a high GPA (It'll be, at …

big current lab machinelearning phd professor publications rankings research school small will work