Greedy Algorithms for Imaging

Greedy Algorithms β€” Fast Sparse Recovery for RF Imaging

Greedy algorithms build the sparse solution iteratively, selecting one or a few atoms per iteration. They are typically much faster than convex relaxation methods (LASSO/FISTA/ADMM) but provide weaker theoretical guarantees. For RF imaging, greedy methods are attractive for real-time applications where the computational budget is tight.

Definition:

Orthogonal Matching Pursuit (OMP) for RF Imaging

OMP builds the support set one element at a time:

  1. Initialize: r(0)=y\mathbf{r}^{(0)} = \mathbf{y}, S(0)=βˆ…S^{(0)} = \emptyset.
  2. Select: jβˆ—=arg⁑max⁑j∣AjHr(t)∣j^* = \arg\max_j |\mathbf{A}_{j}^{H}\mathbf{r}^{(t)}| β€” the pixel most correlated with the residual.
  3. Update support: S(t+1)=S(t)βˆͺ{jβˆ—}S^{(t+1)} = S^{(t)} \cup \{j^*\}.
  4. Project: c^(t+1)=AS(t+1)†y\hat{\mathbf{c}}^{(t+1)} = \mathbf{A}_{S^{(t+1)}}^\dagger \mathbf{y} β€” least squares on the current support.
  5. Update residual: r(t+1)=yβˆ’AS(t+1)c^(t+1)\mathbf{r}^{(t+1)} = \mathbf{y} - \mathbf{A}_{S^{(t+1)}}\hat{\mathbf{c}}^{(t+1)}.
  6. Repeat until βˆ₯rβˆ₯2≀ϡ\|\mathbf{r}\|_2 \leq \epsilon or ∣S∣=smax⁑|S| = s_{\max}.

Cost per iteration: O(MN)O(MN) for the correlation step, O(M∣S∣)O(M|S|) for the least-squares update. Total for ss iterations: O(sMN+s2M)O(sMN + s^2 M).

Definition:

CoSaMP (Compressive Sampling Matching Pursuit)

CoSaMP (Needell and Tropp, 2009) improves on OMP by selecting multiple atoms per iteration and pruning:

  1. Initialize: c^(0)=0\hat{\mathbf{c}}^{(0)} = \mathbf{0}, S(0)=βˆ…S^{(0)} = \emptyset.
  2. Identify: Find the 2s2s largest entries of ∣AHr(t)∣|\mathbf{A}^{H}\mathbf{r}^{(t)}|. Let Ω\Omega be their indices.
  3. Merge: T=S(t)βˆͺΞ©T = S^{(t)} \cup \Omega (at most 3s3s indices).
  4. Estimate: b^=AT†y\hat{\mathbf{b}} = \mathbf{A}_{T}^\dagger\mathbf{y} β€” least squares on the merged support.
  5. Prune: S(t+1)=S^{(t+1)} = indices of the ss largest entries of b^\hat{\mathbf{b}}.
  6. Update: c^(t+1)=b^S(t+1)\hat{\mathbf{c}}^{(t+1)} = \hat{\mathbf{b}}_{S^{(t+1)}}.
  7. Residual: r(t+1)=yβˆ’AS(t+1)c^(t+1)\mathbf{r}^{(t+1)} = \mathbf{y} - \mathbf{A}_{S^{(t+1)}}\hat{\mathbf{c}}^{(t+1)}.

Key advantage over OMP: CoSaMP can correct mistakes β€” atoms added in early iterations can be removed by pruning. OMP commits to each selection permanently.

Definition:

Iterative Hard Thresholding (IHT)

IHT (Blumensath and Davies, 2009) applies hard thresholding after a gradient step:

c(t+1)=Hs ⁣(c(t)+ΞΌAH(yβˆ’Ac(t))),\mathbf{c}^{(t+1)} = \mathcal{H}_s\!\left( \mathbf{c}^{(t)} + \mu\mathbf{A}^{H} (\mathbf{y} - \mathbf{A}\mathbf{c}^{(t)})\right),

where Hs\mathcal{H}_s keeps the ss largest entries and zeros the rest (hard thresholding), and μ≀1/Lf\mu \leq 1/L_f.

IHT vs. ISTA:

  • ISTA uses soft thresholding (β„“1\ell_1 penalty, convex).
  • IHT uses hard thresholding (β„“0\ell_0 constraint, non-convex).
  • IHT has better amplitude accuracy (no shrinkage bias) but requires knowing ss.
  • IHT converges under Ξ΄3s<1/3\delta_{3s} < 1/\sqrt{3}.

For RF imaging, IHT is a useful middle ground between OMP (very fast, committed selections) and FISTA (slower, convex).

OMP for RF Image Reconstruction

Complexity: O(sMN+s2M)O(sMN + s^2 M) total for ss scatterers.
Input: A∈CMΓ—N\mathbf{A} \in \mathbb{C}^{M \times N},
y∈CM\mathbf{y} \in \mathbb{C}^M, sparsity ss or threshold ϡ\epsilon
Initialize: r←y\mathbf{r} \gets \mathbf{y}, Sβ†βˆ…S \gets \emptyset,
c^←0\hat{\mathbf{c}} \gets \mathbf{0}
for t=1,2,…,st = 1, 2, \ldots, s:
// Correlation (matched filter of residual)
h←AHr\mathbf{h} \gets \mathbf{A}^{H} \mathbf{r}
jβˆ—β†arg⁑max⁑j∣hj∣j^* \gets \arg\max_j |h_j|
// Update support
S←Sβˆͺ{jβˆ—}S \gets S \cup \{j^*\}
// Least-squares projection
c^S←AS†y\hat{\mathbf{c}}_S \gets \mathbf{A}_{S}^\dagger \mathbf{y}
// Update residual
r←yβˆ’ASc^S\mathbf{r} \gets \mathbf{y} - \mathbf{A}_{S} \hat{\mathbf{c}}_S
// Early stopping
if βˆ₯rβˆ₯2≀ϡ\|\mathbf{r}\|_2 \leq \epsilon: break
Output: c^\hat{\mathbf{c}}, support SS

The correlation step AHr\mathbf{A}^{H}\mathbf{r} is identical to computing the matched filter image of the residual β€” the same operation as one ISTA gradient step.

OMP vs. LASSO: Speed-Accuracy Tradeoff

Compares OMP, CoSaMP, and LASSO (FISTA) for RF image reconstruction.

Left: Reconstructed images for each algorithm. Right: NMSE vs. computation time, showing the Pareto frontier.

  • OMP: Fastest for small ss but degrades rapidly as sparsity increases.
  • CoSaMP: More robust than OMP with self-correction.
  • LASSO/FISTA: Slowest per iteration but most robust; best for large ss or compressible scenes.

Adjust sparsity to see the crossover point where convex methods outperform greedy ones.

Parameters
10
20

Theorem: OMP Recovery Guarantee for RF Imaging

OMP exactly recovers an ss-sparse signal c\mathbf{c} from y=Ac\mathbf{y} = \mathbf{A}\mathbf{c} in ss iterations if the mutual coherence ΞΌ(A)\mu(\mathbf{A}) satisfies:

ΞΌ(A)<12sβˆ’1.\mu(\mathbf{A}) < \frac{1}{2s - 1}.

More precisely, exact recovery holds if the cumulative coherence function satisfies ΞΌ1(sβˆ’1)+ΞΌ1(s)<1\mu_1(s-1) + \mu_1(s) < 1 (Tropp and Gilbert, 2007).

The condition ensures that each OMP step selects a correct atom: the correlation with the true atom exceeds the correlation with all incorrect atoms combined.

Example: OMP for Real-Time Radar Imaging

Setup: Automotive radar at 77 GHz. Nt=3N_t = 3, Nr=4N_r = 4, Nf=256N_f = 256 (M=3072M = 3072). Scene: 64Γ—6464 \times 64 grid (N=4096N = 4096). Budget: 50 ms per image (20 Hz update).

Method Iterations Time (ms) NMSE (dB)
OMP (s=15s = 15) 15 12 βˆ’15.2-15.2
CoSaMP (s=15s = 15) 10 18 βˆ’17.1-17.1
FISTA (Ξ»=0.03\lambda = 0.03) 30 45 βˆ’19.8-19.8
FISTA (Ξ»=0.03\lambda = 0.03) 50 75 βˆ’21.3-21.3

OMP and CoSaMP meet the real-time requirement; FISTA needs early stopping to fit within the budget.

Greedy vs. Convex Methods for RF Imaging

CriterionGreedy (OMP/CoSaMP/IHT)Convex (LASSO/FISTA/ADMM)
SpeedVery fast for small ssSlower but consistent
RobustnessSensitive to ss estimateRobust (Ξ»\lambda controls sparsity)
Compressible signalsPoorGood (Οƒs\sigma_s bound)
Structured sparsityLimitedTV, group LASSO, etc.
Amplitude accuracyExact (least squares)Biased by Ξ»\lambda (debias needed)
Parameter requirementNeed ss or stopping criterionNeed Ξ»\lambda
RIP guaranteeΞ΄3s<1/3\delta_{3s} < 1/\sqrt{3} (IHT)Ξ΄2s<2βˆ’1\delta_{2s} < \sqrt{2}-1 (LASSO)
,

Common Mistake: Overestimating Sparsity in OMP

Mistake:

Running OMP with smax⁑s_{\max} much larger than the true sparsity. After all true atoms are selected, OMP begins selecting noise atoms, and the least-squares projection contaminates the amplitude estimates of the true atoms.

Correction:

Use the residual norm as a stopping criterion: stop when βˆ₯r(t)βˆ₯2≀MΟƒ\|\mathbf{r}^{(t)}\|_2 \leq \sqrt{M}\sigma. This avoids selecting noise atoms without requiring prior knowledge of ss.

Quick Check

For a radar imaging scene with s=5s = 5 well-separated point scatterers at high SNR\text{SNR}, which algorithm is the best choice for real-time reconstruction?

OMP β€” the small ss and high SNR make it ideal, and it is the fastest option.

FISTA β€” it has stronger theoretical guarantees.

ADMM with TV β€” it produces the best image quality.

⚠️Engineering Note

Practical Guidelines for Algorithm Selection

Decision tree for RF imaging algorithm selection:

  1. Is ss small and known? β†’\to Use OMP or CoSaMP.
  2. Is the scene compressible (power-law decay)? β†’\to Use FISTA.
  3. Does the scene have extended targets? β†’\to Use ADMM with TV.
  4. Is real-time required? β†’\to Use OMP, then refine with FISTA.
  5. Is the scene a mix of point and extended? β†’\to Use ADMM with β„“1\ell_1 + TV.
  6. Are multiple snapshots available? β†’\to Use MMV (Β§Group Sparsity and Joint Recovery (MMV)).

OMP (Orthogonal Matching Pursuit)

Greedy algorithm that builds the sparse support one atom at a time by selecting the atom most correlated with the current residual, then projecting onto the expanded support.

Related: CoSaMP

CoSaMP

Greedy algorithm that selects 2s2s candidates per iteration and prunes back to ss, allowing correction of earlier mistakes.

Related: OMP (Orthogonal Matching Pursuit)

IHT (Iterative Hard Thresholding)

Gradient descent followed by hard thresholding to enforce exact ss-sparsity. No shrinkage bias (unlike ISTA) but requires knowledge of ss.

Key Takeaway

OMP selects one atom per iteration at O(sMN)O(sMN) total cost β€” fastest for very sparse scenes. CoSaMP selects and prunes multiple atoms with self-correction capability. IHT applies hard thresholding for no shrinkage bias but needs ss. Greedy methods are fastest for small ss but degrade for compressible or structured scenes. Practical rule: greedy for real-time with small ss; convex (FISTA/ADMM) for high-quality reconstruction with unknown or large ss.