Intel UHD might refer to the GPU that’s integrated in the CPU (iGPU for short).
Are the displays connected to Nvidia or Intel? See this in the Device Manager, change to View by Connection, then find the displays.
Displays sporadically turning off/on means a bad connection (if you feel that this is not a problem related to sleep – if you are always moving the mouse then it shouldn’t be falling asleep). Therefore try a shorter/different cable – but the dock comes with a short USB-C cable. Are you not using that cable?
Does the problem occur with only one display connected to DisplayPort?
Does the problem occur with only one display connected to HDMI?
A dock doesn’t have drivers. It will have drivers for its component parts. This dock consists of a USB 3.0 hub for the USB ports and built-in USB devices: a USB to Ethernet adapter for the Ethernet port, a USB to SD Card reader for the SD Card slots, a USB Audio device for the audio in/out port, and a DisplayPort 1.2 MST Hub for the display outputs. There may be a driver for the USB hub, a driver for the Ethernet adapter, a driver for the SD Card reader, a driver for the audio port, a driver for any USB devices connected (mouse/keyboard/storage device) and a display driver (Intel or Nvidia) for the MST Hub and any displays connected to the MST Hub. Most of the drivers should be built-in or come from the manufacturer of the individual devices. Connected to the MST Hub is a DisplayPort to VGA adapter and a DisplayPort to HDMI adapter (or they might be built into the hub). I don’t think this is a driver issue. Maybe a driver or firmware update could make it less susceptible to display connection issues by ignoring disconnections that last only a short period of time but probably not.