Polynomial Width is Sufficient for Set Representation with High-dimensional Features