Partial Trace & Reduced States
Partial Trace & Reduced States
You hold one qubit of an entangled pair; your collaborator across the galaxy holds the other. What is your qubit's state? Not a ket — there isn't one. The operation that answers the question is the partial trace, and its output, the reduced density matrix, is the single most important construction in quantum information after the density operator itself. It is how a pure global state becomes a mixed local one, how entanglement is quantified, and (Term 4) how an open system "feels" its environment. It also delivers a clean punchline: the reduced state of a maximally entangled pair is the maximally mixed state — total local ignorance born from perfect global knowledge.
Learning Objectives
After this lesson you will be able to:
- Define the partial trace operationally (on product operators) and by index contraction, and prove it yields a valid density operator.
- Justify the partial trace as the unique operation reproducing local measurement statistics ().
- Prove the headline result: the reduced state of an entangled pure state is mixed, and compute for the Bell state .
- State and apply purification: every mixed state is a reduced state of a pure state on a larger space (Bell example).
- Articulate no-signaling: a local operation on leaves unchanged.
Intuition
A bipartite system lives in (0.1.7). Suppose you only ever touch subsystem : every measurement you can do is of the form — act on , do nothing to . The question "what is the state of ?" can only mean: what single operator on reproduces all of these statistics? That demand — match $\operatorname{Tr}(\rho_{AB}, M_A\otimes I_B)M_A$ — has exactly one answer, and computing it is the partial trace.
The mechanical picture: the full collapses both tensor factors to a number; the partial trace collapses only the factor, leaving an operator on . You are "summing out" — averaging over everything you cannot see. And here is the twist that makes quantum mechanics strange: even when the global is pure (maximal knowledge), summing out an entangled partner injects genuine uncertainty into . Knowledge of the whole does not imply knowledge of the parts. That is the formal signature of entanglement, and the rest of the lesson makes it precise and runs it in code.
Theory
Definition (operational form)
Definition (partial trace). $\operatorname{Tr}_B : \mathcal L(\mathcal H_A\otimes\mathcal H_B) \to \mathcal L(\mathcal H_A)$ is the linear map defined on product operators by
and extended linearly to all operators. The reduced density matrix of is $\rho_A = \operatorname{Tr}B(\rho{AB})\rho_B = \operatorname{Tr}A(\rho{AB})$.
In words: leave the part of each term alone, and replace the part by the number — you take the ordinary trace on the factor only. This matches the preview in 0.1.7. Since every operator on is a linear combination of product operators (these form a basis of $\mathcal L(\mathcal H_A\otimes \mathcal H_B)\operatorname{Tr}_B$ uniquely on everything.
Index form (the computational definition)
Fix orthonormal bases and . Any bipartite operator has matrix elements . The partial trace sums over the repeated index:
Derivation from the operational form. Expand $\rho_{AB} = \sum_{i\mu,j\nu}(\rho_{AB}){i\mu,j\nu},
|i\rangle\langle j|\otimes|\mu\rangle\langle\nu|\operatorname{Tr}B$ termwise:
$\operatorname{Tr}B(|i\rangle\langle j|\otimes|\mu\rangle\langle\nu|) = |i\rangle\langle j|,
\langle\nu|\mu\rangle = \delta{\mu\nu},|i\rangle\langle j|\nu = \mu$ and
sums, giving $\rho_A = \sum{ij}\big(\sum\mu(\rho_{AB}){i\mu,j\mu}\big)|i\rangle\langle j|$, which is
the boxed formula. ∎ Computationally this is a reshape-and-einsum: index $\rho{AB}$ as a
4-tensor and contract — exactly the code below.
The defining property: partial trace reproduces local statistics
The operational definition is not arbitrary; it is forced by physics.
Theorem. is the unique operator on such that, for every observable on ,
This says: predicting any -only measurement from the reduced state gives the same answer as predicting it from the full state with on the untouched side — as it must.
Proof. Existence. For a product operator (with $\sigma\in\mathcal L(\mathcal H_A)\tau\in\mathcal L(\mathcal H_B)\operatorname{Tr}_B(\sigma\otimes\tau) = \sigma,\operatorname{Tr}(\tau)$ directly from the definition. Then
using the mixed-product identity and $\operatorname{Tr}(A\otimes B) = \operatorname{Tr} A \operatorname{Tr} B$ (Appendix E §4–5). By linearity it holds for all . Uniqueness. If two operators satisfied the identity for all , then for all ; taking gives , i.e. the Hilbert–Schmidt norm of the difference is zero, so . ∎
The reduced state is a valid density operator
Proposition. If is a density operator, so is .
Proof. Hermitian: taking commutes with (check on product operators: $(\operatorname{Tr}B(|a\rangle\langle a'|\otimes|b\rangle\langle b'|))^\dagger = (\langle b'|b\rangle|a\rangle\langle a'|)^\dagger = \overline{\langle b'|b\rangle}|a'\rangle\langle a| = \langle b|b'\rangle|a'\rangle\langle a| = \operatorname{Tr}B(|a'\rangle\langle a|\otimes|b'\rangle\langle b|)$), so . Unit trace: $\operatorname{Tr}A(\rho_A) = \operatorname{Tr}{AB}(\rho{AB}) = 1$ (the iterated partial traces equal the full trace). Positive: for any , set $M_A = |\varphi\rangle\langle\varphi|\succeq0\langle\varphi|\rho_A|\varphi\rangle = \operatorname{Tr}A(\rho_A M_A) = \operatorname{Tr}{AB}(\rho{AB},M_A\otimes I_B) \ge 0$ because and (the trace of a product of two PSD operators is ). ∎
The headline result: entangled ⇒ mixed reduced state
Now the payoff. Take a pure global state . When is pure, and when is it mixed?
Theorem. For a bipartite pure state , the reduced state is pure iff is a product state, and mixed iff is entangled. Moreover and have the same nonzero eigenvalues.
Proof (via the Schmidt decomposition). By the Schmidt decomposition (1.4.4, built on the SVD of 0.1.6), write
with , orthonormal and the Schmidt rank. Then
and tracing out (using $\operatorname{Tr}B(|w_k\rangle\langle w_l|) = \langle w_l|w_k\rangle = \delta{kl}$):
So the eigenvalues of are exactly the squared Schmidt coefficients . By the symmetric computation, has the same eigenvalues.
If is a product state, , one : is pure. If is entangled, , so at least two : , hence mixed (Lesson 1's purity bound, with strict inequality whenever two weights are nonzero). ∎
This is striking: a pure carries all the information about , yet — your complete local description — is mixed. The "missing" information is not in or in separately; it lives in the correlations between them. The purity of becomes a measure of entanglement (the basis of the entropy of entanglement , Term 4 and beyond).
Worked engine: for the Bell state
Take the Bell state (1.4.2) . Its density operator is
Rewrite each term as and apply , which replaces by :
The two off-diagonal (coherence) terms are killed by the orthogonality — this is the crux. Summing the survivors:
The reduced state of a maximally entangled pair is maximally mixed: purity , the smallest a qubit allows. Globally you know everything (a single pure ket); locally you know nothing (a fair coin in every basis). The Schmidt coefficients are , confirming maximal entanglement.
Purification
The headline result has a powerful converse.
Theorem (purification). Every density operator on is the reduced state of some pure state on with $\dim\mathcal H_B \ge \operatorname{rank}(\rho_A)|\Psi\rangle\rho_A$.
Proof (construction). Diagonalize (Lesson 1). Pick an orthonormal set in a reference system and define
This is normalized () and pure, and by the Schmidt computation above $\operatorname{Tr}_B(|\Psi\rangle\langle\Psi|) = \sum_k\lambda_k|u_k\rangle\langle u_k| = \rho_A$. ∎
Purification is the formal statement that every mixed state can be seen as part of a larger pure state — classical uncertainty is "really" entanglement with a system you are ignoring (a reference, or, physically, the environment). This "go to the church of the larger Hilbert space" move underpins quantum channels and the Stinespring dilation in Term 4.
Bell example. For we have , . Choosing rebuilds — the Bell state is a purification of the maximally mixed qubit. (Purifications are not unique: any for unitary on gives another, e.g. — the freedom matching Lesson 1's ensemble ambiguity.)
No-signaling
Suppose your distant collaborator performs any physical operation on alone — a unitary , a measurement, even discarding it. Can that change your ? No.
No-signaling (unitary case). If , then is unchanged.
Proof. Using and cyclicity within the factor: on a product operator ,
and linear extension gives $\operatorname{Tr}B((I\otimes U_B)\rho{AB}(I\otimes U_B^\dagger)) = \operatorname{Tr}B(\rho{AB}) = \rho_AB$, the same holds after averaging over 's outcomes — Term 1.3.)
This is why entanglement cannot transmit information: Alice's local description is immune to anything Bob does. Bob can correlate his outcomes with Alice's (that is what entanglement is), but he cannot steer her marginal — consistent with relativity, and the reason the Bell-state correlations of Term 1.4 coexist peacefully with no faster-than-light signaling.
Worked Examples
Example 1 — A product state has a pure reduced state
Take the separable . Then $\rho_{AB} = |+\rangle\langle+|\otimes|0\rangle\langle0|$ is a product operator, so
The reduced state is pure () — exactly as the theorem predicts for a product state (Schmidt rank ). Contrast this with the Bell state, where the same qubit comes out maximally mixed. The difference is entanglement, full stop.
Example 2 — A partially entangled state: tunable purity
Consider (already in Schmidt form: , , orthonormal local bases ). Then
with purity
This interpolates continuously between the extremes: at the state is the product and purity (pure ); at it is and purity (maximally mixed). So the reduced purity is a dial for the amount of entanglement — minimized exactly at the maximally entangled point. We will see in Lesson 3 that this corresponds to the Bloch vector shrinking from the sphere () to the center () as .
Hands-on (Python)
We implement the partial trace via reshape/einsum, confirm for ,
watch the purity drop, and check no-signaling numerically.
import numpy as np
# Big-endian: qubit A = index 0 (leftmost), qubit B = index 1.
ket0 = np.array([1, 0], dtype=complex)
ket1 = np.array([0, 1], dtype=complex)
def partial_trace_B(rho_AB, dimA=2, dimB=2):
"""
Tr_B(ρ_AB): contract the B factor of a bipartite density matrix.
Reshape the (dA·dB)×(dA·dB) matrix into a 4-tensor ρ[i, μ, j, ν]
(row = (i,μ), col = (j,ν)), then sum the diagonal over the B indices μ=ν.
"""
t = rho_AB.reshape(dimA, dimB, dimA, dimB) # ρ[i, μ, j, ν]
return np.einsum('imjm->ij', t) # Σ_μ ρ[i, μ, j, μ]
def partial_trace_A(rho_AB, dimA=2, dimB=2):
"""Tr_A(ρ_AB): contract the A factor instead (sum the A index k=k)."""
t = rho_AB.reshape(dimA, dimB, dimA, dimB) # ρ[k, μ, k, ν]
return np.einsum('kmkn->mn', t) # Σ_k ρ[k, μ, k, ν] -> operator on B
def purity(rho):
return np.trace(rho @ rho).real# Build the Bell state |Φ+⟩ = (|00⟩ + |11⟩)/√2 and its density matrix (big-endian).
ket00 = np.kron(ket0, ket0)
ket11 = np.kron(ket1, ket1)
phi_plus = (ket00 + ket11) / np.sqrt(2)
rho_AB = np.outer(phi_plus, phi_plus.conj())
print("Global purity Tr(ρ_AB²):", round(purity(rho_AB), 6)) # 1.0 (pure global state)
rho_A = partial_trace_B(rho_AB)
print("ρ_A =\n", np.round(rho_A, 6))
print("Is ρ_A = I/2 ?", np.allclose(rho_A, np.eye(2) / 2)) # True
print("Local purity Tr(ρ_A²):", round(purity(rho_A), 6)) # 0.5 -> maximally mixed!
# Entangled pure global state (purity 1) -> mixed local state (purity 1/2).# Tunable entanglement: |Ψθ⟩ = cosθ|00⟩ + sinθ|11⟩ (Worked Example 2).
for theta in [0.0, np.pi/8, np.pi/4]:
psi = np.cos(theta) * ket00 + np.sin(theta) * ket11
rho = np.outer(psi, psi.conj())
rA = partial_trace_B(rho)
print(f"θ={theta:.4f} Tr(ρ_A²)={purity(rA):.4f} "
f"(predicted 1 - ½sin²2θ = {1 - 0.5*np.sin(2*theta)**2:.4f})")
# θ=0 -> 1.0000 (product, pure ρ_A)
# θ=π/4 -> 0.5000 (Bell, maximally mixed) -- minimum purity = max entanglement# No-signaling: any local unitary on B leaves ρ_A untouched.
H = np.array([[1, 1], [1, -1]], dtype=complex) / np.sqrt(2)
S = np.array([[1, 0], [0, 1j]], dtype=complex)
U_B = S @ H # an arbitrary unitary on B
op = np.kron(np.eye(2), U_B) # I_A ⊗ U_B
rho_AB_after = op @ rho_AB @ op.conj().T
rho_A_after = partial_trace_B(rho_AB_after)
print("ρ_A unchanged by Bob's operation?", np.allclose(rho_A_after, rho_A)) # True# Optional cross-check on Braket's LocalSimulator (free): density-matrix simulator
# reproduces ρ_A = I/2 for the Bell state. Deep Braket intro is Term 2 — basic pattern only.
try:
from braket.circuits import Circuit
from braket.devices import LocalSimulator
bell = Circuit().h(0).cnot(0, 1) # prepares (|00⟩+|11⟩)/√2, big-endian
dm_sim = LocalSimulator("braket_dm") # density-matrix local simulator
full_rho = dm_sim.run(bell.density_matrix(), shots=0).result().values[0]
rho_A_braket = partial_trace_B(np.asarray(full_rho))
print("Braket ρ_A = I/2 ?", np.allclose(rho_A_braket, np.eye(2) / 2))
except Exception as e: # SDK not installed / offline — NumPy result already proved it
print("Braket optional check skipped:", type(e).__name__)Why
einsum? Indexing as the 4-tensor makes the index definition literal:'imjm->ij'says "fix free indices on , sum the repeated index ." For qubits, reshape to axes and contract the traced-out subset — the same idea scales (with the usual exponential memory cost of dense simulation, 0.1.7).
Exercises
E1 (easy). Compute and confirm it equals . What does the equality of eigenvalues say in general?
Solution
By the symmetric calculation (trace out instead, using to kill the coherences), . In general and share the same nonzero eigenvalues (the squared Schmidt coefficients), so they have equal purity and equal entropy — the entanglement is a symmetric property of the pair.
E2 (easy). For the product state , compute and and state their purities.
Solution
is a product operator, so $\rho_A = |0\rangle\langle0|,\operatorname{Tr}(|+\rangle\langle+|) = |0\rangle\langle0|\rho_B = |+\rangle\langle+|,\operatorname{Tr}(|0\rangle\langle0|) = |+\rangle\langle+|= 1$. A product state has pure reduced states on both sides (Schmidt rank 1).
E3 (medium). Show that followed by equals the full trace: .
Solution
On a product operator : , then $\operatorname{Tr}A(\cdot) = \langle b'|b\rangle\langle a'|a\rangle\operatorname{Tr}{AB}(|a\rangle\langle a'|\otimes |b\rangle\langle b'|) = \operatorname{Tr}(|a\rangle\langle a'|)\operatorname{Tr}(|b\rangle\langle b'|) = \langle a'|a\rangle\langle b'|b\rangle\rho_{AB}$. ∎ (In particular , reconfirming is unit-trace.)
E4 (medium). Let — the classically correlated (separable, mixed) cousin of the Bell state. Compute . Compare with the Bell-state reduced state and explain why they coincide despite very different global states.
Solution
and $\operatorname{Tr}_B(|11\rangle\langle11|) = |1\rangle\langle1|\rho_A = \tfrac12(|0\rangle\langle0| + |1\rangle\langle1|) = \tfrac12 I\rho_A$. The reduced state cannot tell a quantum-entangled global state from a classically correlated one: both look maximally mixed locally. The distinction lives in correlations and only shows up in joint measurements (e.g. CHSH, Term 1.4) — the local marginal throws that information away. This is exactly the ensemble ambiguity of Lesson 1, viewed through the partial trace.
E5 (hard). Prove that is independent of the orthonormal basis used in the index definition .
Solution
Let be another ONB, related by a unitary on : $|\mu'\rangle = \sum_\nu U_{\nu\mu}|\nu\rangle\sum_\mu\langle i\mu'|\rho_{AB}|j\mu'\rangle = \sum_{\mu,\nu,\sigma} \overline{U_{\nu\mu}}U_{\sigma\mu}\langle i\nu|\rho_{AB}|j\sigma\rangle = \sum_{\nu,\sigma}\big(\sum_\mu U_{\sigma\mu}\overline{U_{\nu\mu}}\big)\langle i\nu|\rho_{AB}|j\sigma\rangle$. The inner sum is , collapsing to — the original expression. So is basis-independent (as it must be, being defined operationally without reference to a basis). ∎
E6 (hard). Use the Schmidt decomposition to prove that any purification of a given is related to the canonical one by an isometry on the purifying system (assuming is large enough). Conclude purifications are unique up to such an isometry.
Solution
Let and $|\Psi'\rangle = \sum_k\sqrt{\lambda_k}|u_k\rangle_A|w_k'\rangle_{B'}\rho_A = \sum_k\lambda_k|u_k\rangle\langle u_k|$ (same -Schmidt vectors and coefficients, since these are fixed by 's spectral decomposition). Define on the relevant subspace by $W|w_k\rangle = |w_k'\rangle{|w_k\rangle}{|w_k'\rangle}W$ is an isometry ( on its support). Then $(I_A\otimes W)|\Psi\rangle = \sum_k\sqrt{\lambda_k}|u_k\rangle W|w_k\rangle = |\Psi'\rangleB$ (a unitary when ). ∎ This is the operator-level statement of "the freedom of purification," and it is the source of Lesson 1's ensemble ambiguity (HJW): different bases for the reference correspond to different ensembles realizing the same .
Checkpoint
- Define the partial trace on product operators and in index form. Which -index gets summed?
- What property forces the partial trace to be exactly this map (and not some other "discard ")?
- State and justify in one line: the reduced state of an entangled pure state is mixed.
- Compute for and give its purity. What does global-pure / local-mixed mean?
- State purification and no-signaling, and say what each is "for."
Answers
- $\operatorname{Tr}B(|a\rangle\langle a'|\otimes|b\rangle\langle b'|) = \langle b'|b\rangle, |a\rangle\langle a'|(\rho_A){ij} = \sum_\mu\langle i\mu|\rho_{AB}|j\mu\rangle$ — the repeated index is summed (set ).
- It is the unique map with $\operatorname{Tr}A(\rho_A M_A) = \operatorname{Tr}{AB}(\rho_{AB}, M_A\otimes I_B)M_A$, i.e. it reproduces all local measurement statistics.
- Schmidt: with eigenvalues squared Schmidt coefficients; entangled means nonzero , so — mixed.
- , purity (maximally mixed). Global-pure/local-mixed: you know the whole pair perfectly yet know nothing about either qubit alone — the information is in the correlations.
- Purification: every mixed is of a pure — for treating mixedness as entanglement with a reference/environment (channels, Term 4). No-signaling: local operations on leave fixed — for ruling out faster-than-light signaling via entanglement.
Further Reading
- [NC] Nielsen & Chuang, §2.4.3 (reduced density operator & partial trace), §2.5 (Schmidt decomposition & purification).
- [Pre] Preskill, Ph219, Ch. 2 — partial trace, the "church of the larger Hilbert space," and no-signaling.
- [Wat] Watrous, Theory of Quantum Information, Ch. 2 — partial trace as a linear map and its characterization.
- [Wil] Wilde, Quantum Information Theory, Ch. 5 — purification and the reduced-state formalism.
- Appendix E §4–5 — trace and Kronecker identities used in the proofs.
← Prev: The Density Operator · Up: Term 1 · Next: The Bloch Ball & Purity →