Note

Go to the end to download the full example code

# Constrained CP decomposition in Tensorly >=0.7

On this page, you will find examples showing how to use constrained CP/Parafac.

## Introduction

Since version 0.7, Tensorly includes constrained CP decomposition which penalizes or constrains factors as chosen by the user. The proposed implementation of constrained CP uses the Alternating Optimization Alternating Direction Method of Multipliers (AO-ADMM) algorithm from [1] which solves alternatively convex optimization problem using primal-dual optimization. In constrained CP decomposition, an auxilliary factor is introduced which is constrained or regularized using an operator called the proximal operator. The proximal operator may therefore change according to the selected constraint or penalization.

Tensorly provides several constraints and their corresponding proximal operators, each can apply to one or all factors in the CP decomposition:

- Non-negativity
non_negative in signature

Prevents negative values in CP factors.

- L1 regularization
l1_reg in signature

Adds a L1 regularization term on the CP factors to the CP cost function, this promotes sparsity in the CP factors. The user chooses the regularization amount.

- L2 regularization
l2_reg in signature

Adds a L2 regularization term on the CP factors to the CP cost function. The user chooses the regularization amount.

- L2 square regularization
l2_square_reg in signature

Adds a L2 regularization term on the CP factors to the CP cost function. The user chooses the regularization amount.

- Unimodality
unimodality in signature

This constraint acts columnwise on the factors

Impose that each column of the factors is unimodal (there is only one local maximum, like a Gaussian).

- Simplex
simplex in signature

This constraint acts columnwise on the factors

Impose that each column of the factors lives on the simplex or user-defined radius (entries are nonnegative and sum to a user-defined positive parameter columnwise).

- Normalization
normalize in signature

Impose that the largest absolute value in the factors elementwise is 1.

- Normalized sparsity
normalized_sparsity in signature

This constraint acts columnwise on the factors

Impose that the columns of factors are both normalized with the L2 norm, and k-sparse (at most k-nonzeros per column) with k user-defined.

- Soft sparsity
soft_sparsity in signature

This constraint acts columnwise on the factors

Impose that the columns of factors have L1 norm bounded by a user-defined threshold.

- Smoothness
smoothness in signature

This constraint acts columnwise on the factors

Favor smoothness in factors columns by penalizing the L2 norm of finite differences. The user chooses the regularization amount. The proximal operator in fact solves a banded system.

- Monotonicity
monotonicity in signature

This constraint acts columnwise on the factors

Impose that the factors are either always increasing or decreasing (user-specified) columnwise. This is based on isotonic regression.

- Hard sparsity
hard_sparsity in signature

This constraint acts columnwise on the factors

Impose that each column of the factors has at most k nonzero entries (k is user-defined).

While some of these constraints (2, 3, 4, 6, 8, 9, 12) require a scalar input as its parameter or regularizer, boolean input could be enough for other constraints (1, 5, 7, 10, 11). Selection of one of these constraints for all mode (or factors) or using different constraints for different modes are both supported.

```
import numpy as np
import tensorly as tl
from tensorly.decomposition import constrained_parafac
import matplotlib.pyplot as plt
np.set_printoptions(precision=2)
# tensor generation
tensor = tl.tensor(np.random.rand(6, 8, 10))
rank = 3
```

## Using one constraint for all modes

Constraints are inputs of the constrained_parafac function, which itself uses the
`tensorly.tenalg.proximal.validate_constraints`

function in order to process the input
of the user. If a user wants to use the same constraint for all modes, an
input (bool or a scalar value or list of scalar values) should be given to this constraint.
Assume, one wants to use unimodality constraint for all modes. Since it does not require
any scalar input, unimodality can be imposed by writing True for unimodality:

```
_, factors = constrained_parafac(tensor, rank=rank, unimodality=True)
```

This constraint imposes that each column of all the factors in the CP decomposition are unimodal:

```
fig = plt.figure()
for i in range(rank):
plt.plot(factors[0][:, i])
plt.legend(['1. column', '2. column', '3. column'], loc='upper left')
```

Constraints requiring a scalar input can be used similarly as follows:

```
_, factors = constrained_parafac(tensor, rank=rank, l1_reg=0.05)
```

The same regularization coefficient l1_reg is used for all the modes. Here the l1 penalization induces sparsity given that the regularization coefficient is large enough.

```
fig = plt.figure()
plt.title('Histogram of 1. factor')
_, _, _ = plt.hist(factors[0].flatten())
fig = plt.figure()
plt.title('Histogram of 2. factor')
_, _, _ = plt.hist(factors[1].flatten())
```

## Using one constraint for some modes

As a second option, constraint can be used for only a few selected modes by using a python dictionary:

```
_, factors = constrained_parafac(tensor, rank=rank, non_negative={0: True, 2: True})
print("1. factor\n", factors[0])
print("2. factor\n", factors[1])
```

```
1. factor
[[4.93 0.69 1.2 ]
[5.06 0.38 0.83]
[4.37 2.9 2.19]
[4.57 1.88 1.41]
[4.55 2.23 1.63]
[5. 0.33 0.85]]
2. factor
[[ 0.27 0.04 0.07]
[ 0.57 0.94 -1.25]
[ 0.38 0.48 -0.48]
[ 0.31 0.28 -0.25]
[ 0.02 -0.54 1.23]
[ 0.43 0.32 -0.59]
[ 0.44 0.31 -0.54]
[ 0.35 0.06 -0.08]]
```

Since only the first and last factors are chosen, entries on the second mode factor could be negative.

## Using a constraint with the different scalar inputs for each mode

One may prefer different scalar value for each mode. It is possible by using a list structure:

```
_, factors = constrained_parafac(tensor, rank=rank, l1_reg=[0.01, 0.02, 0.03])
fig = plt.figure()
plt.title('Histogram of 1. factor')
_, _, _ = plt.hist(factors[0].flatten())
fig = plt.figure()
plt.title('Histogram of 2. factor')
_, _, _ = plt.hist(factors[1].flatten())
fig = plt.figure()
plt.title('Histogram of 3. factor')
_, _, _ = plt.hist(factors[2].flatten())
```

## Using different constraints for each mode

To use different constraint for different modes, the dictionary structure should be preferred:

```
_, factors = constrained_parafac(tensor, rank=rank, non_negative={1:True}, l1_reg={0: 0.01},
l2_square_reg={2: 0.01})
```

In the dictionary, key is the selected mode and value is a scalar value or only True depending on the selected constraint.

```
print("1. factor\n", factors[0])
print("2. factor\n", factors[1])
print("3. factor\n", factors[2])
```

```
1. factor
[[ 1.17e+01 -8.61e+00 -3.51e+00]
[ 1.22e+01 -8.84e+00 -1.28e+01]
[ 1.63e+01 1.26e-02 -1.68e-01]
[ 1.26e+01 -8.54e+00 -1.13e+01]
[ 1.52e+01 -5.13e+00 1.03e+01]
[ 1.12e+01 -1.08e+01 -2.62e+00]]
2. factor
[[0.34 0.39 0.88]
[0.4 0.42 0. ]
[0.39 0.36 0.26]
[0.36 0.2 0.06]
[0.5 0. 0.14]
[0.26 0.57 1.11]
[0.25 0.84 0. ]
[0.34 0.53 0.45]]
3. factor
[[ 0.11 0. -0.01]
[ 0.11 0. -0.03]
[ 0.07 -0.05 -0.02]
[ 0.08 -0.07 0.03]
[ 0.1 -0.03 0.02]
[ 0.07 -0.06 0.01]
[ 0.09 -0.01 -0.01]
[ 0.07 -0.05 0.02]
[ 0.08 -0.03 -0.03]
[ 0.05 -0.07 -0. ]]
```

Thus, first factor will be non-negative, second factor will be regularized by \(0.01\) with \(l_1\) and last factor will be regularized by \(0.01\) with \(l_2^2\).

## References

[1] Huang, Kejun, Nicholas D. Sidiropoulos, and Athanasios P. Liavas. “A flexible and efficient algorithmic framework for constrained matrix and tensor factorization.” IEEE Transactions on Signal Processing 64.19 (2016): 5052-5065. (Online version)

**Total running time of the script:** ( 0 minutes 4.047 seconds)