Explicit construction for unitary extensions of completely positive and trace preserving (CPTP) maps?

The isometry $Y:\mathcal H\rightarrow \mathcal H_E \otimes \mathcal H$ is $$ Y=\left(\begin{array}{c} A_1 \\ \vdots \\ A_K \end{array}\right) = \sum_k |k\rangle \otimes A_k\ . $$ Clearly, $$ \mathrm{tr}_E(Y\rho Y^\dagger) = \sum_{kl} \mathrm{tr}(|k\rangle\langle l|) A_k\rho A_l^\dagger = \sum_k A_k \rho A_k^\dagger \ , $$ as desired. Moreover, $Y$ is an isometry, $Y^\dagger Y=I$, i.e., its columns are orthonormal, which follows from the condition $\sum_k A_k^\dagger A_k=I$ (i.e., the map is trace preserving).

Now if you want to obtain a unitary which acts on $|0\rangle\langle 0|\otimes \rho$ the same way $Y$ acts on $\rho$, you have to extend the matrix $Y$ to a unitary by adding orthogonal column vectors. For instance, you can pick linearly independent vectors from your favorite basis and orthonormalize. (Clearly, $U$ is highly non-unique, as its action on environment states other than $|0\rangle$ is not well defined.)


This is just to complement the other answer by providing the explicit construction.

We have a map $\Phi(\rho)=\sum_k A_k \rho A_k^\dagger$ and we want to find a unitary $\mathcal U$, acting on a bigger system+environment space, such that: $$\newcommand{\tr}{\operatorname{tr}} \tr_E\big(\mathcal U(\rho\otimes|0\rangle\!\langle0|)\mathcal U^\dagger\big) = \Phi(\rho), \tag1 $$ where $|0\rangle$ is some (pure) state of the environment.

Let us write the map $\mathcal U$ as $$\mathcal U=\sum_{ijkl} u_{ij,kl} |i\rangle\!\langle j|\otimes |k\rangle\!\langle l|, $$ where $i, j$ refer to the system and $k, l$ to the environment degrees of freedom.

The action of $\mathcal U$ over $\rho\otimes|0_E\rangle\!\langle0_E|$ gives \begin{aligned} \mathcal U(\rho\otimes|0_E\rangle\!\langle0_E|) \mathcal U^\dagger &= u_{ij,kl}u^*_{mn,pq} |i\rangle\!\langle j|\rho|n\rangle\!\langle m| \otimes |k\rangle\!\langle l|0\rangle\!\langle 0| q\rangle\!\langle p| \\ &= u_{ij,k0}u^*_{mn,p0} \rho_{jn} |i\rangle\!\langle m|\otimes |k\rangle\!\langle p|, \end{aligned} where every index is assumed to be implicitly summed over.

Taking the trace over the environment now gives: \begin{aligned} \tr_E\Big[\mathcal U(\rho\otimes|0_E\rangle\!\langle0_E|) \mathcal U^\dagger \Big] &= \sum_{ijkmn} u_{ij,k0}u^*_{mn,k0} \rho_{jn} |i\rangle\!\langle m| \\ &= \sum_k A^k_{ij}\rho_{jn} (A^{k})^\dagger_{nm} |i\rangle\!\langle m| \\ &= \sum_k A^k \rho\, (A^k)^\dagger. \end{aligned} The above tells us that the action of the unitary evolution $\mathcal U$ on the enlarged space, when looking only at the effective evolution over the reduced state, amounts to a mapping of the form $\rho\mapsto \sum_k A^k\rho A^{k\dagger}$, as we expected.

In other words, (1) is verified with any $\mathcal U$ satisfying $(\mathcal U)_{ij,k0}=A^k_{ij}$, or, equivalently, $$(\mathbb 1\otimes \langle k\rvert)\mathcal U(\mathbb1\otimes\lvert 0\rangle) = A^k.$$

An intuitive way to understand the above constraint is by considering the environment as the first space (so the opposite of the way we have written above). Note that this doesn't change anything of relevance, it is just a different way to represent the same things. However, when the environment is the first space, we can think of $\mathcal U$ as composed of a number of blocks (submatrices), each one characterizing the action of $\mathcal U$ on the system when looking at a specific input-output combination over the environment.

The above reasoning can be pictured as following:

enter image description here

where in this example the environment has dimension 3.

With the above picture in mind, the condition $\sum_k A^{k\dagger}A^k=\mathbb 1$ is also straightforwardly understood: it is equivalent to asking the columns of $\mathcal U$ corresponding to the input state of the environment (that is, the columns in the middle block in the image above) to be a set of orthonormal vectors.