Not sure why this happens to you. I have HA with several dozens WiFi devices and I have only 2 devices (one relay, one sensor) that disconnect regularly, they have both poor WiFi signal, one in a basement and one far from the AP. Almost all are on 2.4 GHz, not by choice, but they work well.